Example #1
0
def api_distribute_map(
    xs: Union[Sequence[oneflow_api.BlobDesc], oneflow_api.BlobDesc],
    f: Callable[[oneflow_api.BlobDesc, oneflow_api.BlobDesc], oneflow_api.BlobDesc],
    axis: int = 0,
) -> Tuple[oneflow_api.BlobDesc]:
    func = enable_if.unqiue([distribute_map])
    return func(xs, f, axis=axis)
Example #2
0
def api_distribute_map(
    xs: Union[Sequence[remote_blob_util.BlobDef], remote_blob_util.BlobDef],
    f: Callable[[remote_blob_util.BlobDef, remote_blob_util.BlobDef],
                remote_blob_util.BlobDef],
    axis: int = 0,
) -> Tuple[remote_blob_util.BlobDef]:
    func = enable_if.unqiue([distribute_map])
    return func(xs, f, axis=axis)