oneflow.advanced

Advanced features

oneflow.advanced.distribute_add(xs: Sequence[oneflow_api.BlobDesc], name: Optional[str] = None) → oneflow_api.BlobDesc

oneflow.advanced.distribute_clone(x: oneflow_api.BlobDesc, name: Optional[str] = None) → Tuple[oneflow_api.BlobDesc]

oneflow.advanced.distribute_concat(xs: Sequence[oneflow_api.BlobDesc], axis: int = 0, name: Optional[str] = None) → oneflow_api.BlobDesc

oneflow.advanced.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]

oneflow.advanced.distribute_split(x: oneflow_api.BlobDesc, axis: int = 0, name: Optional[str] = None) → Tuple[oneflow_api.BlobDesc]