MCPcopy Create free account

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

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

↓ 7 callersMethodidentity
(*args, **kwargs)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:1053
↓ 7 callersFunctionisSignLessTensorOrOther
oneflow/ir/lib/OneFlow/Conversion/OneFlowToTosa.cpp:79
↓ 7 callersMethodmake_shared
oneflow/core/intrusive/shared_ptr.h:44
↓ 7 callersMethodmode
Returns the mode of the distribution.
python/oneflow/distributions/distribution.py:103
↓ 7 callersMethodmut_autograd_meta
oneflow/core/framework/tensor.h:449
↓ 7 callersFunctionnode_replace_args
(args, name2node)
python/oneflow/framework/infer_compiler/with_fx_graph.py:144
↓ 7 callersMethodoffload
oneflow/core/framework/tensor.cpp:167
↓ 7 callersMethodport
oneflow/core/control/ctrl_server.h:34
↓ 7 callersMethodset_is_leaf
oneflow/core/framework/tensor.h:445
↓ 7 callersMethodsrc_actor_id
oneflow/core/lazy/actor/actor_message.h:45
↓ 7 callersMethodstacked_error
oneflow/core/common/maybe.h:191
↓ 7 callersMethodstacked_error
oneflow/core/common/error.h:137
↓ 7 callersMethodthread_uid
oneflow/core/framework/stream.h:49
↓ 6 callersFunctionAddAccumulateFunctionNode
oneflow/core/autograd/autograd_engine.cpp:520
↓ 6 callersMethodAddEntry
Add i-th element into this subset
oneflow/core/auto_parallel/binary_set.cpp:61
↓ 6 callersMethodAddNdSbpSignature4OpName
oneflow/core/job/job_builder.cpp:447
↓ 6 callersMethodAllocPinned
oneflow/core/ep/cpu/cpu_device.cpp:70
↓ 6 callersMethodBindEdgeWithProducedRegst
oneflow/core/graph/task_node.cpp:300
↓ 6 callersMethodBorrowTensorName
oneflow/core/framework/tensor.cpp:40
↓ 6 callersFunctionBroadcast
oneflow/core/framework/op_interpreter/eager_local_op_interpreter.cpp:220
↓ 6 callersMethodBuildCtrlRegstDesc
oneflow/core/graph/task_node.cpp:285
↓ 6 callersFunctionCheckDataShape
oneflow/user/ops/one_embedding_ops.cpp:221
↓ 6 callersFunctionCheckDataType
oneflow/user/ops/one_embedding_ops.cpp:244
↓ 6 callersFunctionCheckInplace
oneflow/core/ep/common/primitive/util.h:41
↓ 6 callersFunctionCheckLearningRateShape
oneflow/user/ops/multi_tensor_model_update_ops.cpp:53
↓ 6 callersMethodCompute
oneflow/core/vm/instruction.cpp:52
↓ 6 callersMethodCount
oneflow/api/cpp/framework/shape.cpp:57
↓ 6 callersFunctionCreateAvgOpKernelCache
oneflow/user/kernels/avg_pool_kernel.cpp:26
↓ 6 callersFunctionCreatePoolOpKernelCache
oneflow/user/kernels/max_pool_kernel.cpp:26
↓ 6 callersFunctionCudnnSPZeroPtr
oneflow/core/device/cudnn_util.cpp:231
↓ 6 callersFunctionDfsTopoGraphTraversal
oneflow/core/job_rewriter/pass_util.cpp:31
↓ 6 callersMethodFindId4NdSbp
Find corresponding id for Nd sbp
oneflow/core/auto_parallel/boxing_collector.cpp:185
↓ 6 callersMethodFinish
oneflow/core/profiler/event.cpp:35
↓ 6 callersMethodForEachEdge
oneflow/core/graph/graph.h:289
↓ 6 callersMethodForEachFrontRegst
oneflow/core/lazy/actor/register_slot.cpp:136
↓ 6 callersMethodFreePinned
oneflow/core/ep/cpu/cpu_device.cpp:76
↓ 6 callersFunctionGenerateComputeTaskStreamId
oneflow/core/graph/task_stream_id.h:24
↓ 6 callersMethodGet
oneflow/core/embedding/cache.h:64
↓ 6 callersMethodGetCommForDeviceAndStreamName
oneflow/core/job/eager_nccl_comm_manager.cpp:134
↓ 6 callersFunctionGetCurTime
oneflow/core/common/util.h:153
↓ 6 callersMethodGetDevice
oneflow/core/hardware/node_device_descriptor.cpp:276
↓ 6 callersMethodGetDeviceCount
oneflow/core/ep/cpu/cpu_device_manager.cpp:38
↓ 6 callersMethodGetDeviceManager
oneflow/core/ep/common/device_manager_registry.cpp:159
↓ 6 callersFunctionGetDeviceTagOfTensor
oneflow/core/framework/op_interpreter/lazy_op_interpreter.cpp:124
↓ 6 callersFunctionGetEmbeddingUpdateSbp
oneflow/user/ops/one_embedding_ops.cpp:261
↓ 6 callersMethodGetParallelDescSymbol
oneflow/core/framework/instructions_builder.cpp:219
↓ 6 callersFunctionGetParallelHierarchyNumAxes
oneflow/core/job/job_builder.cpp:31
↓ 6 callersFunctionGetPhysicalShape
oneflow/core/framework/tensor_impl.cpp:210
↓ 6 callersMethodGetPrevRankInRing
oneflow/core/job/rank_group.cpp:114
↓ 6 callersFunctionGetPtr
oneflow/ir/lib/OneFlow/Passes.cpp:140
↓ 6 callersFunctionGetRoot
oneflow/core/graph/inplace_lbi_graph.cpp:63
↓ 6 callersFunctionGetSbpSignatures
oneflow/user/ops/assign_op.cpp:36
↓ 6 callersFunctionGetTensorByteSize
oneflow/user/kernels/nccl_logical_fusion_kernel.cpp:62
↓ 6 callersMethodGetTotalBufferSize
oneflow/user/kernels/model_update_kernels.cpp:51
↓ 6 callersFunctionGetTransferCost
oneflow/core/framework/sbp_infer_util.cpp:752
↓ 6 callersMethodGetWeightedCost
oneflow/core/auto_parallel/sbp_node.h:148
↓ 6 callersFunctionGetWindowedOutputSize
oneflow/core/operator/operator_util.cpp:63
↓ 6 callersFunctionGlobalJobBuildAndInferCtxMgr
oneflow/core/job/job_build_and_infer_ctx_mgr.cpp:96
↓ 6 callersFunctionHobIsSendAndRecvRegistered
oneflow/user/kernels/communicate_util.h:27
↓ 6 callersFunctionIdxOffset
oneflow/user/ops/nn_util.cpp:75
↓ 6 callersMethodInit
oneflow/core/vm/op_call_instruction_policy.cpp:206
↓ 6 callersFunctionIsBoolDataType
oneflow/core/common/data_type.cpp:21
↓ 6 callersFunctionIsErr
oneflow/maybe/maybe.h:194
↓ 6 callersMethodIsOk
oneflow/maybe/maybe.h:73
↓ 6 callersFunctionIsSplitSbp
oneflow/core/functional/impl/slice_boxing_functor.cpp:35
↓ 6 callersFunctionLocalTensorTo
oneflow/core/functional/impl/array_functor.cpp:3087
↓ 6 callersFunctionMachineId4ActorId
oneflow/core/graph/task_id.cpp:73
↓ 6 callersFunctionMakeBroadcastSbpParallel
oneflow/core/job/sbp_parallel.cpp:76
↓ 6 callersFunctionMakeInitialScope
oneflow/core/framework/scope_util.cpp:62
↓ 6 callersFunctionMakeInplaceProposalFn
oneflow/user/kernels/image_object_preprocess_kernels.cpp:431
↓ 6 callersMethodMaybeGetPtr
oneflow/core/vm/symbol_storage.h:122
↓ 6 callersMethodMoveToDstBack
oneflow/core/intrusive/list.h:71
↓ 6 callersMethodMutOpTransactionCommit
oneflow/core/job/job_builder.cpp:362
↓ 6 callersMethodMutShapeView4ArgNameAndIndex
oneflow/core/kernel/user_kernel.cpp:497
↓ 6 callersMethodName
oneflow/core/hardware/cuda_device_descriptor_class.cpp:57
↓ 6 callersFunctionNdSbpSignatureToSbpSignature
oneflow/core/job/sbp_parallel.cpp:274
↓ 6 callersFunctionNewMatmulPrimitive
oneflow/user/kernels/group_conv_kernel.cpp:30
↓ 6 callersMethodNumAxes
oneflow/api/cpp/framework/shape.cpp:55
↓ 6 callersMethodNumUniqueDiffIndicesPtr
oneflow/user/kernels/model_update_kernels.cpp:60
↓ 6 callersMethodOpConf4OpName
oneflow/core/job/job_builder.cpp:161
↓ 6 callersFunctionOpTypeName4OpNode
oneflow/core/job_rewriter/quantization_aware_training.cpp:122
↓ 6 callersMethodParallelConf4Lbi
oneflow/core/job/job_builder.cpp:165
↓ 6 callersFunctionParseNdSbpFromLongString
oneflow/core/job/sbp_parallel.cpp:197
↓ 6 callersFunctionPenalty4PartialInConsumer
oneflow/core/framework/sbp_infer_util.cpp:53
↓ 6 callersFunctionPyDTypeCheck
DType
oneflow/api/python/functional/common.cpp:230
↓ 6 callersFunctionPyLongSequenceCheck
oneflow/api/python/functional/common.cpp:98
↓ 6 callersFunctionPySbpParallelSequenceCheck
SBP list
oneflow/api/python/functional/common.cpp:300
↓ 6 callersFunctionPyUnicodeToStdString
oneflow/extension/stack/python/stack_getter.cpp:48
↓ 6 callersFunctionPyUnpackDType
oneflow/api/python/functional/common.cpp:231
↓ 6 callersFunctionPyUnpackSbpParallel
oneflow/api/python/functional/common.cpp:295
↓ 6 callersMethodRecordEvent
oneflow/core/ep/cpu/cpu_stream.cpp:29
↓ 6 callersFunctionSafeAdd2D
oneflow/user/kernels/grid_sample_kernel_util.h:296
↓ 6 callersFunctionSbpToNdSbp
oneflow/core/framework/nd_sbp.cpp:170
↓ 6 callersFunctionScopeSymbolId4Lbi
oneflow/core/job_rewriter/autograd.cpp:37
↓ 6 callersMethodSend
oneflow/core/transport/transport.cpp:165
↓ 6 callersMethodSendMsg
oneflow/core/lazy/actor/actor_message_bus.cpp:24
↓ 6 callersFunctionSeqFor
oneflow/core/thread/thread_runtime.h:43
↓ 6 callersFunctionSharedLibPaths
oneflow/ir/lib/OneFlow/OKL/Kernel/JITEngine.cpp:40
↓ 6 callersMethodShrinkAllMem
oneflow/core/vm/virtual_machine.cpp:191
↓ 6 callersFunctionStrCat
oneflow/core/common/str_util.h:32
← previousnext →1,001–1,100 of 28,456, ranked by callers