MCPcopy Create free account

hub / github.com/Oneflow-Inc/oneflow / functions

Functions28,456 in github.com/Oneflow-Inc/oneflow

↓ 34 callersMethodAddLbi
oneflow/core/graph/task_node.h:194
↓ 34 callersFunctionDeviceTag4DeviceType
oneflow/core/framework/to_string.cpp:23
↓ 34 callersMethodFront
oneflow/core/lazy/actor/register_slot.cpp:70
↓ 34 callersMethodNumAxes
oneflow/core/ndarray/xpu_shape.h:43
↓ 34 callersMethodapply
r"""Adds the forward pre-hook that enables pruning on the fly and the reparametrization of a tensor in terms of the original tensor an
python/oneflow/nn/utils/prune.py:427
↓ 34 callersMethoddebug
r"""Open or close debug mode of the graph. If in debug mode, logs of computation graph building infos or warnings will be printed. Ot
python/oneflow/nn/graph/graph.py:481
↓ 34 callersFunctionri
(indices)
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:229
↓ 34 callersMethodvalues
Return an iterable of the ModuleDict values.
python/oneflow/nn/utils/container.py:246
↓ 33 callersMethodBuild
Explicitly complete the construction of the builtin op Returns: the completed builtin op
python/oneflow/ops/stateful_ops.py:68
↓ 33 callersFunctionIsIntegralDataType
oneflow/core/common/data_type.cpp:31
↓ 33 callersFunctionPbMessage2TxtString
oneflow/core/common/protobuf.cpp:54
↓ 33 callersFunctiong
(x)
python/oneflow/test/exceptions/throw_error.py:23
↓ 33 callersMethodstorage_offset
oneflow/core/framework/tensor.h:83
↓ 32 callersFunctionAutoMemcpy
oneflow/core/kernel/kernel_util.cpp:27
↓ 32 callersMethodGetXShape5D
oneflow/user/kernels/max_pool_kernel_util.cpp:104
↓ 32 callersMethodOutput
Set output blob of op Args: output_name (str): name of output blob num (int, optional): Defaults to 1. Retu
python/oneflow/ops/stateful_ops.py:54
↓ 32 callersMethod__print
r"""Do print according to info level.
python/oneflow/nn/graph/graph.py:630
↓ 32 callersMethodis_eager
oneflow/core/framework/tensor.h:66
↓ 32 callersFunctionis_in_memory
(tensor)
python/oneflow/test/expensive/_test_remat.py:37
↓ 31 callersFunctionLocalToGlobal
oneflow/core/functional/impl/global_cast.cpp:415
↓ 31 callersMethod_shallow_repr
(self)
python/oneflow/nn/graph/graph.py:611
↓ 31 callersMethodend
oneflow/api/python/caster/size.h:38
↓ 31 callersFunctionlog
oneflow/core/common/bfloat16_math.h:32
↓ 31 callersMethodmutable_sbp_signature
oneflow/core/operator/user_op.cpp:450
↓ 31 callersMethodnext
oneflow/core/intrusive/list_hook.h:31
↓ 31 callersMethodregister_buffer
r""" register_buffer(name, tensor, persistent=True) Adds a buffer to the module. This is typically used to register
python/oneflow/nn/modules/module.py:277
↓ 31 callersFunctiontorch2oflow
(mod, *args, **kwargs)
python/oneflow/framework/infer_compiler/transform/builtin_transform.py:162
↓ 30 callersMethodBegin
oneflow/core/intrusive/list.h:49
↓ 30 callersFunctionIsUserOpWithTypeName
oneflow/core/job_rewriter/pass_util.cpp:74
↓ 30 callersFunctionNarrow
oneflow/core/framework/tensor_methods.cpp:515
↓ 30 callersMethodNewNode
oneflow/core/graph/graph.h:314
↓ 30 callersFunctionPyTensor_New
oneflow/api/python/framework/tensor.cpp:854
↓ 30 callersMethodToDimVector
oneflow/core/common/shape_view.cpp:22
↓ 30 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_block.py:284
↓ 30 callersFunction_cpu_global_tensor
(tensor)
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:54
↓ 30 callersMethoddata
oneflow/core/common/tensor_buffer.h:110
↓ 30 callersMethodis_floating_point
oneflow/core/framework/dtype.cpp:152
↓ 30 callersMethodmin
oneflow/core/common/bfloat16.h:283
↓ 30 callersMethodoutput_size
oneflow/core/framework/user_op_conf.cpp:78
↓ 29 callersFunctionDefaultAutoGenerator
oneflow/core/framework/random_generator.cpp:136
↓ 29 callersMethodErase
oneflow/core/intrusive/list.h:110
↓ 29 callersMethodIsIntegral
oneflow/core/common/scalar.h:89
↓ 29 callersMethodIsOk
oneflow/core/common/maybe.h:187
↓ 29 callersMethodeval
(self)
python/oneflow/test_utils/automated_test_util/generators.py:95
↓ 29 callersMethodgetOpName
oneflow/ir/include/OneFlow/OneFlowOpTraits.h:60
↓ 29 callersFunctionglobal_broadcast_consec
Generate a arithmetic progression with given size and start value.
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:65
↓ 29 callersMethodmut_head
Setters
oneflow/core/intrusive/list_test.cpp:307
↓ 29 callersMethodremove
(self)
python/oneflow/utils/hooks.py:39
↓ 29 callersMethodshape
oneflow/api/python/caster/size.h:30
↓ 29 callersMethodtensor_storage
oneflow/core/framework/tensor.h:81
↓ 28 callersFunctionDeviceAndStreamInferFn
oneflow/user/ops/comm_net_device_infer_util.h:34
↓ 28 callersFunctionExpand
oneflow/core/framework/tensor_methods.cpp:394
↓ 28 callersFunctionSetInputArgModifierMutable
oneflow/user/ops/model_update_ops.cpp:339
↓ 28 callersFunctionUnsqueeze
oneflow/core/framework/tensor_methods.cpp:220
↓ 28 callersFunctioncompare_result
(a, b, rtol=1e-5, atol=1e-8)
python/oneflow/test/tensor/test_complex.py:69
↓ 28 callersMethodref_cnt
oneflow/core/intrusive/ref.h:33
↓ 28 callersMethodrun
tools/oneflow-tblgen/op_schema_emitter.cpp:110
↓ 27 callersFunctionGetInDim
oneflow/core/operator/operator_util.cpp:45
↓ 27 callersFunctionHashCombine
oneflow/maybe/utility.h:69
↓ 27 callersFunctionSwitchCase
oneflow/core/common/switch_func.h:24
↓ 27 callersFunction_global_math_op_grad_grad_impl
(test_case, op_name, placement, sbp)
python/oneflow/test/modules/test_global_math_op_higher_derivative.py:24
↓ 27 callersFunction_is_raw_type
(value, raw_type)
python/oneflow/framework/args_tree.py:21
↓ 27 callersFunction_test_math_op_grad_grad_impl
(test_case, op_name)
python/oneflow/test/modules/test_math_op_higher_derivative.py:24
↓ 27 callersMethodadd_module
r""" add_module(name, module) Adds a child module to the current module. The module can be accessed as an attribute
python/oneflow/nn/modules/module.py:252
↓ 27 callersMethoddevice
oneflow/api/cpp/framework/tensor.cpp:62
↓ 27 callersMethodis_complex
oneflow/core/framework/dtype.cpp:117
↓ 27 callersMethodmap
oneflow/core/common/optional.h:410
↓ 27 callersMethodreset
oneflow/core/framework/mutable_attr_map.h:53
↓ 26 callersFunctionCastToPyObject
oneflow/api/python/functional/common.h:205
↓ 26 callersMethodGetSoleRegst
oneflow/core/graph/task_node.cpp:418
↓ 26 callersFunctionIsFloatingDataType
oneflow/core/common/data_type.cpp:40
↓ 26 callersMethodLookup
oneflow/core/lazy/actor/light_actor.cpp:129
↓ 26 callersFunctionMakeHostMemCase
oneflow/core/memory/memory_case_util.cpp:101
↓ 26 callersMethodSet
oneflow/core/ndarray/xpu_shape.h:48
↓ 26 callersFunctionTranspose
oneflow/core/framework/tensor_methods.cpp:735
↓ 26 callersMethodWrite
oneflow/core/persistence/tee_persistent_log_stream.cpp:42
↓ 26 callersMethod__init__
(self, alpha: float = 1.0, inplace: bool = False)
python/oneflow/nn/modules/activation.py:254
↓ 26 callersFunctioncompare_result
(test_case, a, b, rtol=1e-5, atol=1e-8)
python/oneflow/test/modules/test_fused_glu.py:123
↓ 26 callersMethodlength
oneflow/user/kernels/pocketfft_hdronly.h:2510
↓ 26 callersMethodparse_args
oneflow/extension/stack/stacktrace.h:689
↓ 26 callersFunctionrandom_dtype
(seq_names)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:1437
↓ 26 callersFunctionspawn_shell
(cmd: str = None)
ci/test/distributed_run.py:63
↓ 25 callersMethodByteSizeOfBlobBody
oneflow/core/register/blob.h:121
↓ 25 callersMethodDecrease
oneflow/core/common/blocking_counter.cpp:31
↓ 25 callersFunctionGetModule
oneflow/ir/lib/OneFlow/OKL/Kernel/LauncherState.cpp:41
↓ 25 callersFunctionGetValidMaxCopyCost
oneflow/core/framework/sbp_infer_util.cpp:746
↓ 25 callersMethodIsFloatingPoint
oneflow/core/common/scalar.h:90
↓ 25 callersFunctionRegisterBoxingFunction
oneflow/core/boxing/eager_boxing_interpreter.h:59
↓ 25 callersFunctionTestEnumerateNdSbpIn2OutSignatures
oneflow/user/ops/reshape_user_op_util_test.cpp:68
↓ 25 callersMethodTryPushBackRegst
oneflow/core/lazy/actor/register_slot.cpp:37
↓ 25 callersMethod__contains__
(self, key: str)
python/oneflow/nn/utils/container.py:218
↓ 25 callersMethodbegin
oneflow/api/python/caster/size.h:37
↓ 25 callersMethoddata_type
oneflow/core/kernel/blob_tensor_view.cpp:31
↓ 25 callersMethodenable_reuse_mem
oneflow/core/job/job_desc.h:53
↓ 25 callersMethodparallel_num
oneflow/core/framework/op_expr.cpp:494
↓ 25 callersMethodrank
oneflow/core/graph/stream_id.h:42
↓ 25 callersMethodretain_grad
oneflow/core/framework/tensor.h:426
↓ 25 callersMethodto
Performs dtype and/or device conversion on `self.data`. It has similar signature as :meth:`oneflow.Tensor.to`, except optional argume
python/oneflow/nn/utils/rnn.py:162
↓ 25 callersMethodupdate
(self, event)
python/oneflow/profiler/events.py:47
↓ 24 callersFunctionGetPhysicalShape
oneflow/core/operator/operator.cpp:1654
← previousnext →301–400 of 28,456, ranked by callers