MCPcopy Create free account

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

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

↓ 3 callersFunctionConcatSplitDataContent
oneflow/core/kernel/boxing_kernel.cpp:170
↓ 3 callersMethodConstruct
oneflow/maybe/optional.h:76
↓ 3 callersFunctionConstructPbRpf
oneflow/core/kernel/boxing_kernel.cpp:43
↓ 3 callersFunctionConsumeOpsToFunc
oneflow/ir/lib/OneFlow/Passes.cpp:778
↓ 3 callersFunctionConvBNSiLU
(out, in_channels, channels, kernel=1, stride=1, pad=0, num_group=1)
python/oneflow/test/expensive/pytorch_rexnet.py:61
↓ 3 callersFunctionConvComputationCost
Example for conv2d: ComputationCost = ((k*k + k*k-1)*c + c-1 + bias?1:0) * out_channel * out_width * out_height * batch_size = (2*k*k*c - 1 + bias?1:
oneflow/user/ops/conv_op.cpp:128
↓ 3 callersFunctionConvertNorm
oneflow/core/common/data_type_converter.h:386
↓ 3 callersFunctionConvertProtobufMsg2Json
oneflow/user/summary/summary_converter.h:83
↓ 3 callersFunctionConvertUserOp
oneflow/ir/lib/OneFlow/OneFlowInferReturnTypes.cpp:52
↓ 3 callersFunctionConvertUserOpAttributes
oneflow/ir/lib/OneFlow/UserOpConversion.cpp:169
↓ 3 callersFunctionCopyBetweenLocalTensorAndNumpy
oneflow/api/python/utils/tensor_utils.h:124
↓ 3 callersFunctionCpuBroadcast
oneflow/core/ccl/ccl.cpp:52
↓ 3 callersFunctionCpuNdarrayCopy
oneflow/core/ndarray/cpu_ndarray_copy.h:24
↓ 3 callersFunctionCreateEpBackendDeviceAllocator
oneflow/core/vm/ep_stream_policy.cpp:33
↓ 3 callersFunctionCreateIndexedSlicesUpdateOpKernelCache
oneflow/user/kernels/model_update_kernels.cpp:94
↓ 3 callersFunctionCreateNcclComm
oneflow/core/job/eager_nccl_comm_manager.cpp:57
↓ 3 callersMethodCreateOrGetAllocator
oneflow/core/vm/remat/allocator.cpp:747
↓ 3 callersMethodData_YouAreNotAllowedToCallThisFuncOutsideThisFile
generate a temporary object to allow `const auto& x = optval().value()` where `optval()` is a function call which returns a temporary Optional
oneflow/core/common/optional.h:401
↓ 3 callersMethodDeallocateBlobDataPtr
oneflow/core/eager/eager_blob_object.cpp:125
↓ 3 callersMethodDecreasePoolSizeByBase
oneflow/core/common/tensor_buffer.cpp:278
↓ 3 callersMethodDecreaseRefCount
oneflow/core/intrusive/ref.h:63
↓ 3 callersMethodDefaultDataType
oneflow/core/job/job_desc.h:49
↓ 3 callersMethodDeinitJob
oneflow/core/job/collective_boxing/scheduler.cpp:147
↓ 3 callersFunctionDeviceMax
oneflow/user/kernels/clip_by_value_kernel.h:34
↓ 3 callersFunctionDeviceMin
oneflow/user/kernels/clip_by_value_kernel.h:25
↓ 3 callersFunctionDivUp
oneflow/core/thread/thread_runtime.h:45
↓ 3 callersMethodDropMaxLayer
Drop down the maximum layer with the minimum layer form consumer
oneflow/core/auto_parallel/sbp_node.cpp:565
↓ 3 callersMethodDropTributaryLayer
Drop down the maximum layer with the minimum layer form consumer
oneflow/core/auto_parallel/sbp_node.cpp:808
↓ 3 callersFunctionDropoutImpl
oneflow/core/functional/impl/nn_functor.cpp:3171
↓ 3 callersMethodDumpLogicalBlobDesc
oneflow/core/graph/op_graph.cpp:532
↓ 3 callersMethodEmplace
oneflow/maybe/variant.h:299
↓ 3 callersMethodEmplace
oneflow/maybe/optional.h:203
↓ 3 callersFunctionEncodeDeviceIdToInt64
oneflow/core/device/device_id.cpp:31
↓ 3 callersFunctionEncodeTaskIdToInt64
oneflow/core/graph/task_id.cpp:52
↓ 3 callersMethodEvalInNbhCost
Compute the cost between this node and adjacent nodes with a lower order
oneflow/core/auto_parallel/sbp_node.cpp:412
↓ 3 callersMethodEvalOutNbhCost
oneflow/core/auto_parallel/sbp_node.cpp:388
↓ 3 callersFunctionFilterNdSbpByLogicalShape
Judge whether an NdSbp could be applied on a tensor with given logical shape True means this NdSbp is not valid.
oneflow/core/framework/sbp_infer_util.cpp:882
↓ 3 callersMethodFindEdgeWithNode
oneflow/core/auto_parallel/sbp_node.cpp:839
↓ 3 callersFunctionForEachAggregatedParamGroup
oneflow/core/job_rewriter/autograd.cpp:254
↓ 3 callersMethodForEachDataAndCtrlOutNode
oneflow/core/graph/op_graph.cpp:475
↓ 3 callersMethodForEachLbi
oneflow/core/register/register_desc.cpp:112
↓ 3 callersMethodForEachMutRequestEntryForIdsInJob
oneflow/core/job/collective_boxing/request_store.h:102
↓ 3 callersMethodForEachRank
oneflow/core/job/rank_group.cpp:126
↓ 3 callersMethodForward
oneflow/api/cpp/framework/graph.cpp:208
↓ 3 callersFunctionFwGetSbpFn
oneflow/user/ops/normalization_op.cpp:99
↓ 3 callersFunctionFwInputArgModifyFn
oneflow/user/ops/normalization_op.cpp:75
↓ 3 callersFunctionGenCartesianProduct
(sets)
python/oneflow/test/graph/test_util.py:28
↓ 3 callersFunctionGenConnInfoKey
oneflow/core/comm_network/ibverbs/ibverbs_comm_network.cpp:34
↓ 3 callersFunctionGenPortKey
oneflow/core/comm_network/epoll/epoll_comm_network.cpp:71
↓ 3 callersFunctionGenVariableOutputLbn
oneflow/core/job_rewriter/adam_optm.cpp:73
↓ 3 callersFunctionGenerateAdamHelperVariableOpConf
oneflow/core/job_rewriter/adam_optm.cpp:78
↓ 3 callersFunctionGet1DZeroVariableOpConf
oneflow/core/job_rewriter/quantization_aware_training.cpp:129
↓ 3 callersMethodGetActorContext
oneflow/core/lazy/actor/light_actor.cpp:664
↓ 3 callersMethodGetAsyncError
oneflow/core/ep/cuda/cuda_stream.cpp:240
↓ 3 callersFunctionGetComputationCost
Theoretically computation cost of matrix multiplication is the products of the number of matrix and first dimension of matrix a, second dimension of m
oneflow/user/ops/matmul_op.cpp:86
↓ 3 callersFunctionGetCublasScalarParameter
oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp:68
↓ 3 callersFunctionGetCudnnVersion
oneflow/core/ep/cuda/cuda_device_manager_factory.cpp:36
↓ 3 callersFunctionGetCwd
oneflow/core/common/process_state.h:32
↓ 3 callersFunctionGetCyclicBoxingDebugString
oneflow/core/framework/placement_sbp_util.cpp:483
↓ 3 callersFunctionGetDualNdSbp
oneflow/core/framework/nd_sbp.cpp:54
↓ 3 callersFunctionGetErrorMsgOfSearchedOp
oneflow/core/framework/user_op_registry_manager.cpp:67
↓ 3 callersMethodGetFileSize
oneflow/core/persistence/posix/posix_file_system.cpp:160
↓ 3 callersFunctionGetFormatedSerializedError
oneflow/core/common/maybe.h:273
↓ 3 callersFunctionGetGeluSbp
oneflow/user/ops/gelu_op.cpp:33
↓ 3 callersMethodGetMinCost
Get the minimum element in Cost
oneflow/core/auto_parallel/sbp_node.cpp:582
↓ 3 callersFunctionGetNcclUniqueIdRpcKey
oneflow/core/job/eager_nccl_comm_manager.cpp:33
↓ 3 callersFunctionGetNdSbpValidTransformationAxisSequence
oneflow/core/framework/placement_sbp_util.cpp:467
↓ 3 callersFunctionGetOpNames
oneflow/core/job_rewriter/autotick.cpp:329
↓ 3 callersFunctionGetPtrType
use this func to union the ptr type in this conversion phase.
oneflow/ir/lib/OneFlow/OKL/Conversion/OKLToLLVM.cpp:52
↓ 3 callersFunctionGetRandomSeedForRank
NOTE: Get random seed in current rank, and ensure that it will have same seed between broadcast sbp and it will be different between split sbp. It wi
oneflow/user/kernels/random_seed_util.cpp:35
↓ 3 callersMethodGetReadyRegstSize
oneflow/core/lazy/actor/register_slot.cpp:20
↓ 3 callersFunctionGetRegistry
oneflow/ir/include/OneFlow/OKL/Kernel/LauncherState.h:32
↓ 3 callersFunctionGetRegstDescCount
oneflow/core/lazy/actor/light_actor.cpp:186
↓ 3 callersMethodGetResult
oneflow/core/framework/user_op_registry.h:152
↓ 3 callersFunctionGetRunningActorCountKeyByJobId
oneflow/core/job/runtime_context.cpp:39
↓ 3 callersFunctionGetSbpSignature
oneflow/core/operator/interface_op_util.cpp:27
↓ 3 callersFunctionGetSbpSignatures4Conv
oneflow/user/ops/conv_op.cpp:91
↓ 3 callersFunctionGetSbpSignatures4DeConv
oneflow/user/ops/deconv_op.cpp:92
↓ 3 callersMethodGetScopeSymbol
oneflow/core/framework/instructions_builder.cpp:223
↓ 3 callersFunctionGetSizeInSlice
[start, end)
oneflow/user/kernels/slice_kernel.cpp:33
↓ 3 callersFunctionGetStageIdHint
oneflow/core/job_rewriter/fix_pipeline_stage_id_pass.cpp:65
↓ 3 callersFunctionGetTensorBufferPoolSize
oneflow/core/common/tensor_buffer.cpp:198
↓ 3 callersFunctionGetThisThreadGlobalId
oneflow/core/thread/thread_global_id.cpp:33
↓ 3 callersFunctionGetUniqueMemCaseId
oneflow/core/memory/memory_case_util.cpp:79
↓ 3 callersMethodGetVariableNode
oneflow/core/job_rewriter/optimizer_placement_optimization_pass.cpp:51
↓ 3 callersFunctionGlobalSessionUtilMutex
oneflow/core/framework/session_util.cpp:22
↓ 3 callersMethodGlobalTensorMeta4ArgNameAndIndex
oneflow/user/kernels/stateful_opkernel.cpp:75
↓ 3 callersFunctionGridSamplerUnnormalize
oneflow/user/kernels/grid_sample_kernel_util.h:79
↓ 3 callersFunctionGridSamplerUnnormalizeSetGrad
oneflow/user/kernels/grid_sample_kernel_util.h:95
↓ 3 callersMethodHasPinnedDevice
oneflow/core/ep/include/allocation_options.h:34
↓ 3 callersMethodHasProducerTaskId4RegstDescId
oneflow/core/register/register_manager.cpp:277
↓ 3 callersMethodIncreasePoolSizeByBase
oneflow/core/common/tensor_buffer.cpp:271
↓ 3 callersMethodIncreaseRefCount
oneflow/core/intrusive/ref.h:62
↓ 3 callersFunctionInferDataType_
oneflow/user/ops/deconv_op.cpp:87
↓ 3 callersFunctionInferLocalSignature
oneflow/core/operator/operator.cpp:1533
↓ 3 callersMethodInferLocalSignatureIf
oneflow/core/operator/operator.cpp:953
↓ 3 callersMethodInferLogicalOutBlobDescsIf
oneflow/core/operator/operator.cpp:305
↓ 3 callersMethodInferNdSbpSignatureIf
oneflow/core/operator/operator.cpp:893
↓ 3 callersMethodInferParallelSignatureIf
oneflow/core/operator/operator.cpp:118
↓ 3 callersMethodInit
Construct a boxing collector with given maximum number of axis
oneflow/core/auto_parallel/boxing_collector.cpp:139
← previousnext →1,801–1,900 of 28,456, ranked by callers