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)
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)