Python unqiue Beispiele

Beispiel #1

Datei: distribute_ops.py Projekt: zhenlin-work/oneflow

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)

Beispiel #2

Datei: distribute_ops.py Projekt: Sodu-Qinming/Oneflow

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)