MCPcopy Create free account

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

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

↓ 24 callersFunctionRoundUp
oneflow/core/common/util.h:163
↓ 24 callersMethodToProto
oneflow/core/common/range.cpp:25
↓ 24 callersMethoddevice_type
oneflow/core/job/parallel_desc.h:58
↓ 24 callersMethodmemory_format
oneflow/core/framework/tensor_impl.cpp:170
↓ 24 callersMethodmemory_format
oneflow/core/register/blob.h:57
↓ 24 callersMethodptr
oneflow/core/ndarray/xpu_var_ndarray.h:49
↓ 24 callersMethodref_cnt
oneflow/core/intrusive/list_test.cpp:48
↓ 24 callersMethodresource
(self)
python/oneflow/framework/multi_client_session.py:78
↓ 24 callersMethodsbp_signature
oneflow/core/operator/operator.cpp:1048
↓ 24 callersMethodset_stride
oneflow/core/eager/eager_blob_object.cpp:67
↓ 24 callersMethodtype
(self)
python/oneflow/nn/graph/graph_block.py:66
↓ 23 callersMethodAdd
oneflow/core/job/cluster_instruction.cpp:54
↓ 23 callersMethodEqualsIgnoringHierarchy
oneflow/core/job/parallel_desc.cpp:225
↓ 23 callersMethodInsert
oneflow/core/intrusive/skiplist.h:65
↓ 23 callersFunctionSqueeze
oneflow/core/framework/tensor_methods.cpp:307
↓ 23 callersMethodcurrent_seed
oneflow/core/framework/random_generator.cpp:54
↓ 23 callersMethoddevice_type
oneflow/core/boxing/ccl_boxing_function.cpp:33
↓ 23 callersFunctionget_numpy
(tensor, indices)
python/oneflow/test/tensor/test_tensor_indexing2.py:476
↓ 23 callersFunctionisnan
oneflow/core/common/bfloat16.h:251
↓ 22 callersFunctionGetGeneratorForLazyOrGlobal
oneflow/user/kernels/random_seed_util.cpp:73
↓ 22 callersMethodGetMemory
oneflow/core/auto_parallel/sbp_graph.cpp:117
↓ 22 callersMethodGetPtrOrThrow
oneflow/core/common/maybe.h:108
↓ 22 callersFunctionIsIndexDataType
oneflow/core/common/data_type.cpp:77
↓ 22 callersFunction_compare_with_eager
( test_case, *, batch_size=8, acc=1, norm_type=2.0, max_norm=1.0, device="cuda",
python/oneflow/test/graph/test_graph_clip_grad_norm.py:205
↓ 22 callersFunction_test_dequantize
(test_case, num_bits, shape, group_dim, group_size)
python/oneflow/test/modules/test_groupwise_quantization.py:118
↓ 22 callersMethodclose
Closes access to the shared memory from this instance but does not destroy the shared memory block.
python/oneflow/multiprocessing/shared_memory/__init__.py:63
↓ 22 callersMethodcompute_local_dep_object
oneflow/core/framework/tensor_impl.cpp:116
↓ 22 callersMethoddevice_type
oneflow/core/kernel/user_kernel.cpp:649
↓ 22 callersMethodparse
oneflow/ir/lib/OneFlow/OneFlowOps.cpp:286
↓ 22 callersMethodregst_desc
oneflow/core/register/register.h:45
↓ 22 callersMethodstream_type
oneflow/core/framework/stream.h:48
↓ 22 callersMethodtask_proto
oneflow/core/graph/plan_task_graph.h:39
↓ 21 callersMethodDebugStr
oneflow/core/common/shape.cpp:80
↓ 21 callersMethodEmplaceBack
oneflow/core/intrusive/list.h:98
↓ 21 callersFunctionGetCurInferCtx
oneflow/core/job/job_build_and_infer_ctx_mgr.cpp:105
↓ 21 callersFunctionIsViewApplicable
oneflow/core/framework/tensor_methods.cpp:42
↓ 21 callersMethodLaunch
oneflow/core/kernel/kernel.cpp:52
↓ 21 callersFunctionSbpToString
oneflow/core/framework/nd_sbp.cpp:128
↓ 21 callersMethodSetOutputStride
oneflow/user/kernels/stateful_opkernel.cpp:403
↓ 21 callersFunctionTxtString2PbMessage
oneflow/core/common/protobuf.cpp:64
↓ 21 callersFunctionUnaryPrimitiveExists
oneflow/user/kernels/elementwise_primitive_kernel.h:118
↓ 21 callersFunctionUniqueStr
oneflow/core/framework/id_util.cpp:21
↓ 21 callersMethod__init__
(self, a, b)
python/oneflow/test_utils/automated_test_util/generators.py:151
↓ 21 callersFunctioncos
oneflow/core/common/bfloat16_math.h:37
↓ 21 callersMethoddecode
(self, *args, **kwargs)
python/oneflow/framework/infer_compiler/with_oneflow_compile.py:344
↓ 20 callersFunctionBinaryPrimitiveExists
oneflow/user/kernels/elementwise_primitive_kernel.h:130
↓ 20 callersFunctionCurrentRankSync
oneflow/core/vm/vm_util.cpp:48
↓ 20 callersMethodGetOpTimeShape
oneflow/core/operator/operator.cpp:486
↓ 20 callersMethodInitedDone
oneflow/core/lazy/actor/register_slot.cpp:94
↓ 20 callersFunctionNew
Methods
oneflow/core/functional/impl/global_cast.cpp:84
↓ 20 callersFunctionParseIntegerFromEnv
oneflow/core/common/util.cpp:161
↓ 20 callersFunctionStackedError
oneflow/maybe/maybe.h:182
↓ 20 callersMethodTensorTupleIndex4ArgNameAndIndex
oneflow/core/framework/arg_tuple.cpp:58
↓ 20 callersMethodWaitDone
oneflow/core/framework/transport_util.h:36
↓ 20 callersFunction_compare_graph_lr_scheduler_with_eager
(test_case, **kwargs)
python/oneflow/test/graph/test_graph_lr_scheduler.py:74
↓ 20 callersMethodallocation_type
oneflow/core/register/register.h:58
↓ 20 callersMethodchain_id
oneflow/core/graph/task_node.h:59
↓ 20 callersFunctionconsec
Generate a arithmetic progression with given size and start value.
python/oneflow/test/tensor/test_tensor_indexing2.py:35
↓ 20 callersMethoddevice_id
oneflow/core/framework/device.h:45
↓ 20 callersMethodenable_debug_mode
oneflow/core/job/resource_desc.cpp:61
↓ 20 callersMethodextend
Appends modules from a Python iterable to the end of the list. Arguments: modules (iterable): iterable of modules to
python/oneflow/nn/utils/container.py:174
↓ 20 callersMethodget_kind
oneflow/ir/oneflow-extension/include/PyAst/Ast.h:68
↓ 20 callersMethodhalf
r""" half() Casts all floating point parameters and buffers to ``half`` datatype. .. note:: This method
python/oneflow/nn/modules/module.py:1662
↓ 20 callersMethodinput_size
oneflow/core/framework/user_op_conf.cpp:72
↓ 20 callersMethodnd_sbp_signature
oneflow/core/operator/operator.cpp:1053
↓ 20 callersMethodprev
oneflow/core/intrusive/list_hook.h:30
↓ 20 callersMethodregst
oneflow/core/lazy/actor/actor_message.cpp:93
↓ 20 callersMethodsub
oneflow/core/ep/include/primitive/fast_integer_math.h:63
↓ 20 callersMethodwait
oneflow/user/kernels/pocketfft_hdronly.h:615
↓ 19 callersFunctionAddHash
oneflow/core/common/hash.h:30
↓ 19 callersMethodAddRegst
oneflow/core/graph/task_node.cpp:432
↓ 19 callersMethodGetOpRegistryResult
oneflow/core/framework/user_op_registry_manager.cpp:49
↓ 19 callersMethodGetVmStream
oneflow/core/vm/virtual_machine.cpp:391
↓ 19 callersMethodIsCurSlotReady
oneflow/core/lazy/actor/register_slot.h:34
↓ 19 callersFunctionIsInDebugMode
oneflow/core/common/env_var/debug_mode.h:26
↓ 19 callersFunctionPyTensor_Check
oneflow/api/python/framework/tensor.h:38
↓ 19 callersFunctionSyncAccessTensorWithTimeOut
oneflow/core/framework/tensor_util.cpp:29
↓ 19 callersMethodToProto
oneflow/core/graph/task_node.cpp:239
↓ 19 callersFunction_test_activation_module
(test_case, act_type)
python/oneflow/test/modules/test_global_activation.py:102
↓ 19 callersMethodgetArgument
oneflow/ir/lib/Transform/TransformDialectInterpreter.cpp:41
↓ 19 callersMethodhas_input
oneflow/core/framework/user_op_conf.cpp:64
↓ 19 callersMethodmodules
r""" modules() -> Iterator["Module"] Returns an iterator over all modules in the network. Yields: Module
python/oneflow/nn/modules/module.py:684
↓ 19 callersMethodmut_buffer
oneflow/core/vm/instruction.h:44
↓ 19 callersMethodmut_stream
Setters
oneflow/core/vm/instruction.h:133
↓ 19 callersMethodmut_stream_policy
oneflow/core/vm/instruction.cpp:78
↓ 19 callersFunctionnp_allclose_with_shape
(a, b, *args, **kwargs)
python/oneflow/test/modules/test_ddp.py:29
↓ 18 callersMethodCopyFrom
oneflow/core/ndarray/cpu_var_ndarray.h:70
↓ 18 callersMethodDelOps
oneflow/core/job/job_builder.cpp:313
↓ 18 callersFunctionEncodeStreamIdToInt64
oneflow/core/graph/stream_id.cpp:48
↓ 18 callersMethodForEachNodeOnOutEdge
oneflow/core/graph/node.h:98
↓ 18 callersFunctionGenArgList
(arg_dict)
python/oneflow/test/expensive/test_util.py:38
↓ 18 callersFunctionGetCudnnDataType
oneflow/core/device/cudnn_util.cpp:23
↓ 18 callersMethodHas
oneflow/core/framework/attr_map.cpp:117
↓ 18 callersFunctionIsKeyFound
oneflow/core/job_rewriter/pass_util.h:29
↓ 18 callersFunctionIsOk
oneflow/maybe/maybe.h:193
↓ 18 callersFunctionLoadSymbol
oneflow/core/platform/lib/ibv_wrapper.cpp:45
↓ 18 callersMethodTopoForEachNode
oneflow/core/graph/graph.h:251
↓ 18 callersMethodWaitUntilCntEqualZero
oneflow/core/job/runtime_context.cpp:32
↓ 18 callersFunctionWithinBounds3D
oneflow/user/kernels/grid_sample_kernel_util.h:276
↓ 18 callersFunctionallocated_memory
(device, include_test_placeholder=False)
python/oneflow/test/expensive/_test_remat.py:44
← previousnext →401–500 of 28,456, ranked by callers