oneflow.experimental

Experimental features

class oneflow.experimental.CustomOpModule(op_module_name, module_path='')

__init__(op_module_name, module_path='')

Initialize self. See help(type(self)) for accurate signature.

build_load()

cpp_def()

cpp_kernel()

gpu_kernel()

py_api()

py_kernel()

oneflow.experimental.custom_op_module

alias of oneflow.python.ops.util.custom_op_module.CustomOpModule

oneflow.experimental.dynamic_binary_concat(input_blob_list: Sequence[oneflow_api.BlobDesc], source_blob: oneflow.python.framework.input_blob_def.ArgBlobDef, source_sbp: str = 'S:0', name: Optional[str] = None) → oneflow_api.BlobDesc

oneflow.experimental.dynamic_binary_split(x: oneflow.python.framework.input_blob_def.ArgBlobDef, base_shift: int = 2, out_num: int = 2, name: Optional[str] = None) → List[oneflow_api.BlobDesc]

oneflow.experimental.eager_assign_121(ref, value)

oneflow.experimental.enable_typing_check(val: bool = True) → None

enable typing check for global_function

oneflow.experimental.get_interface_blob_value(op_name)

oneflow.experimental.indexed_slices_reduce_sum(indices: oneflow.python.framework.input_blob_def.ArgBlobDef, values: oneflow.python.framework.input_blob_def.ArgBlobDef, name: Optional[str] = None) → Tuple[oneflow_api.BlobDesc]

oneflow.experimental.logical_slice(x: oneflow_api.BlobDesc, slice_tup_list: Sequence[Tuple[int, int, int]], name: Optional[str] = None) → oneflow_api.BlobDesc

oneflow.experimental.logical_slice_assign(x: oneflow_api.BlobDesc, value: oneflow_api.BlobDesc, slice_tup_list: Sequence[Tuple[int, int, int]], name: Optional[str] = None) → oneflow_api.BlobDesc

oneflow.experimental.set_interface_blob_value(op_name, ndarray)

oneflow.experimental.square_sum(x: oneflow_api.BlobDesc, name: Optional[str] = None) → oneflow_api.BlobDesc

oneflow.experimental.ssp_variable_proxy(var: oneflow_api.BlobDesc, buffer_size: int = 1, name=None) → Tuple[oneflow_api.BlobDesc, oneflow_api.BlobDesc]

return ref_blob, value_blob

oneflow.experimental.unique_with_counts(x: oneflow.python.framework.input_blob_def.ArgBlobDef, out_idx: oneflow.python.framework.dtype.dtype = <class 'oneflow.python.framework.dtype.int32'>, name: Optional[str] = None) → Tuple[oneflow_api.BlobDesc]