MCPcopy Create free account

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

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

⨍ Functions 28,456◇ Types & classes 6,969↳ Endpoints 179
↓ 4 callersMethodLaunchGraph
oneflow/core/ep/cuda/cuda_stream.cpp:319
↓ 4 callersMethodLoadSnapshot
oneflow/core/embedding/persistent_table.cpp:653
↓ 4 callersFunctionLocalFS
oneflow/core/persistence/file_system.cpp:187
↓ 4 callersMethodMainByteSize4OneRegst
oneflow/core/register/runtime_register_desc.cpp:107
↓ 4 callersMethodMakeChildScopeProto
oneflow/core/job/scope.cpp:88
↓ 4 callersFunctionMakeEagerBlobObjectList
oneflow/core/framework/nn_graph.cpp:87
↓ 4 callersFunctionMakePredicatorIsSafeToDelete
oneflow/core/job_rewriter/pass_util.cpp:57
↓ 4 callersMethodMakeSplitSignatureListBuilder
oneflow/core/job/sbp_signature_builder.cpp:96
↓ 4 callersMethodMaybeGet
oneflow/core/vm/symbol_storage.h:110
↓ 4 callersFunctionMemReusedAlgorithmAllocateByOrder
oneflow/core/job/intra_job_mem_sharing_util.h:52
↓ 4 callersFunctionMemcpyPrimitiveExists
oneflow/user/kernels/matmul_kernels.cpp:112
↓ 4 callersFunctionMetaInfoConsistencyCheck
oneflow/core/framework/consistency_check.cpp:226
↓ 4 callersFunctionMutName2BoxingChecker
oneflow/core/boxing/eager_boxing_interpreter.cpp:58
↓ 4 callersMethodMutSrcNode4Ibn
oneflow/core/graph/op_graph.cpp:130
↓ 4 callersFunctionMutThreadLocalTensorMetaCheckDepth
oneflow/core/framework/tensor_rpc_util.cpp:154
↓ 4 callersMethodName
oneflow/ir/oneflow-extension/include/PyAst/Ast.h:403
↓ 4 callersFunctionNdSbpDimReduce
oneflow/core/framework/sbp_infer_util.cpp:418
↓ 4 callersFunctionNdSbpIsAllPartialSum
oneflow/core/boxing/naive_1_to_p_boxing.cpp:27
↓ 4 callersFunctionNewPrimitiveFromHandlers
oneflow/core/ep/common/primitive/util.h:251
↓ 4 callersFunctionNewRandomSeed
oneflow/core/common/util.h:137
↓ 4 callersMethodNewUniqueOpNameByFunctionalOpConf
oneflow/core/job/job_build_and_infer_ctx.cpp:1244
↓ 4 callersMethodOnExecutionContextSetup
oneflow/core/ep/cuda/cuda_stream.cpp:209
↓ 4 callersMethodOnExecutionContextTeardown
oneflow/core/ep/cuda/cuda_stream.cpp:215
↓ 4 callersMethodOneElemSize
oneflow/core/kernel/boxing_kernel.cpp:95
↓ 4 callersMethodPlacementNew
oneflow/core/memory/memory_allocator.h:51
↓ 4 callersMethodPopBack
oneflow/core/intrusive/list.h:115
↓ 4 callersMethodProcessJob
oneflow/ir/oneflow-translate/lib/OneFlow/MLIROneFlowTranslation.cpp:546
↓ 4 callersMethodPushSplitInfo
oneflow/user/kernels/slice_kernel.cpp:59
↓ 4 callersMethodPushStack
oneflow/maybe/error.h:260
↓ 4 callersMethodPut
oneflow/core/job_rewriter/pass_util.h:70
↓ 4 callersFunctionPyUnpackLong
int64_t
oneflow/api/python/functional/common.cpp:410
↓ 4 callersFunctionPyUnpackSbpParallelSequence
oneflow/api/python/functional/common.cpp:303
↓ 4 callersFunctionPyUnpackScalar
oneflow/api/python/functional/common.cpp:151
↓ 4 callersMethodReadFully
oneflow/core/persistence/persistent_in_stream.cpp:113
↓ 4 callersFunctionRegsiteredSessionIds
oneflow/core/framework/session_util.cpp:27
↓ 4 callersMethodRegstDesc4BnInOp
oneflow/core/graph/exec_graph.h:60
↓ 4 callersFunctionReluGrad
oneflow/user/kernels/normalization_kernel.cpp:191
↓ 4 callersFunctionRequireSameSbp
Judge whether we need the same SBP for both producer and consumer
oneflow/core/auto_parallel/sbp_util.cpp:27
↓ 4 callersMethodReserve
oneflow/core/common/tensor_buffer.cpp:93
↓ 4 callersMethodReset
oneflow/core/vm/sync_access_instruction_policy.h:71
↓ 4 callersMethodReset
oneflow/maybe/optional.h:64
↓ 4 callersMethodResetState
oneflow/core/job_rewriter/job_pass.h:80
↓ 4 callersMethodSaveSnapshot
oneflow/core/embedding/persistent_table.cpp:700
↓ 4 callersFunctionScheduleUntilVMEmpty
oneflow/core/vm/virtual_machine.cpp:165
↓ 4 callersMethodSeparatedHeaderByteSize4OneRegst
oneflow/core/register/runtime_register_desc.cpp:123
↓ 4 callersFunctionSetAlgo4Perf
oneflow/core/device/cudnn_conv_util.cpp:55
↓ 4 callersMethodSetIsDynamic4ArgNameAndIndex
oneflow/user/kernels/stateful_opkernel.cpp:459
↓ 4 callersMethodSetIsMatchedPred
oneflow/core/kernel/kernel_registration.h:43
↓ 4 callersMethodSetPinnedDevice
oneflow/core/ep/include/allocation_options.h:46
↓ 4 callersFunctionSetShuttingDown
oneflow/core/framework/shut_down_util.cpp:35
↓ 4 callersFunctionSharedPtrShapeEqual
oneflow/core/job_rewriter/logical_chain_pass.cpp:120
↓ 4 callersFunctionShiftNegativeAxisIfNeed
oneflow/user/ops/layer_norm_op.cpp:23
↓ 4 callersFunctionSoftSyncNNGraphBuffers
oneflow/core/framework/nn_graph.cpp:915
↓ 4 callersMethodSoftmax
oneflow/core/ep/include/primitive/softmax.h:29
↓ 4 callersMethodStackedError
oneflow/maybe/maybe.h:134
↓ 4 callersFunctionStorage4NdSbp
Compute storage per device for given NdSbp
oneflow/core/framework/sbp_infer_util.cpp:852
↓ 4 callersFunctionStraightenOpGraph
Straighten the whole op graph
oneflow/core/auto_parallel/auto_memory.cpp:541
↓ 4 callersMethodSummarizeCost
oneflow/core/auto_parallel/sbp_node.cpp:175
↓ 4 callersFunctionTEST
oneflow/core/ndarray/cpu_concat_var_ndarray_test.cpp:23
↓ 4 callersMethodTensorDesc4ArgNameAndIndex
oneflow/core/kernel/user_kernel.cpp:651
↓ 4 callersFunctionTensorToDenseElementsAttr
oneflow/ir/lib/OneFlow/OneFlowSupport.cpp:162
↓ 4 callersFunctionTestKeyValueStore
oneflow/core/embedding/key_value_store_test.cpp:49
↓ 4 callersMethodThread
oneflow/core/thread/thread.cpp:29
↓ 4 callersFunctionThreadLocalScopeStack
oneflow/core/framework/scope_util.cpp:37
↓ 4 callersFunctionToShape
oneflow/core/functional/impl/global_cast.cpp:111
↓ 4 callersFunctionToString
oneflow/core/common/to_string.h:25
↓ 4 callersMethodToString
oneflow/api/python/autograd/function_node.cpp:31
↓ 4 callersMethodTotalElemNum
oneflow/core/kernel/boxing_kernel.cpp:97
↓ 4 callersMethodTruncate
oneflow/core/embedding/posix_file.h:75
↓ 4 callersMethodTryGetParallelId
oneflow/core/job/parallel_desc.cpp:189
↓ 4 callersFunctionUnaryFold
oneflow/ir/lib/OneFlow/OneFlowOpFolders.cpp:65
↓ 4 callersFunctionVerifyAMPList
oneflow/core/job_rewriter/auto_mixed_precision.cpp:31
↓ 4 callersMethodVisualStr
oneflow/core/graph/node.h:61
↓ 4 callersMethodVisualStr
oneflow/core/graph/task_node.cpp:197
↓ 4 callersFunction__Init__
methods
oneflow/core/intrusive/head_free_list_test.cpp:45
↓ 4 callersMethod__UnsafeMoveTo__
oneflow/core/intrusive/shared_ptr.h:85
↓ 4 callersMethod__all_global
(test_case, input, placement, sbp)
python/oneflow/test/modules/test_global_cast.py:1305
↓ 4 callersMethod__build_io
(self, io_type, build_func, *args, **kwargs)
python/oneflow/nn/graph/graph.py:1797
↓ 4 callersMethod__flatten_io
(self, io_type, *args, **kwargs)
python/oneflow/nn/graph/graph.py:1914
↓ 4 callersMethod__init__
(self)
python/oneflow/test/modules/test_parameters_grouping.py:33
↓ 4 callersMethod__init__
(self)
python/oneflow/test/modules/test_ddp.py:42
↓ 4 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytorch_pvt.py:27
↓ 4 callersMethod__init__
Parameters ---------- block (nn.Module): Bottleneck class. - For SENet154: SEBottleneck - For SE-ResN
python/oneflow/test/expensive/pytorch_senet.py:167
↓ 4 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytorch_resmlp.py:28
↓ 4 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_with_global.py:230
↓ 4 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_random_seed.py:66
↓ 4 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_zero.py:135
↓ 4 callersMethod__init__
(self, modules: Optional[Iterable[T]] = None)
python/oneflow/nn/utils/container.py:107
↓ 4 callersMethod__map_io_lite
(self, func, *args, **kwargs)
python/oneflow/nn/graph/graph.py:1907
↓ 4 callersFunction_assert_true
(test_case, value1, value2)
python/oneflow/test/modules/test_higher_derivative_softmax.py:24
↓ 4 callersFunction_assert_true
(test_case, value1, value2)
python/oneflow/test/modules/test_global_higher_derivative_softmax.py:24
↓ 4 callersFunction_compare_with_nn_cross_entropy_loss
( test_case, pred, label, pred_sbp=None, label_sbp=None )
python/oneflow/test/graph/test_graph_sparse_softmax_cross_entropy.py:58
↓ 4 callersFunction_compute_nparams_toprune
r"""Since amount can be expressed either in absolute value or as a percentage of the number of units/channels in a tensor, this utility functi
python/oneflow/nn/utils/prune.py:1252
↓ 4 callersMethod_conv_forward
(self, x, weight, bias)
python/oneflow/nn/modules/conv.py:243
↓ 4 callersFunction_get_device_index
r"""Gets the device index from :attr:`device`, which can be a torch.device object, a Python integer, or ``None``. If :attr:`device` is a flow
python/oneflow/cuda/_utils.py:32
↓ 4 callersMethod_get_from_states
(self, name, states_name)
python/oneflow/nn/graph/proxy.py:507
↓ 4 callersFunction_get_indexes
(device)
python/oneflow/test/tensor/test_tensor_part_3.py:25
↓ 4 callersFunction_inspect_rand_op_and_args
(rand_op, **kwargs)
python/oneflow/test/modules/test_random_generator_and_seed.py:33
↓ 4 callersFunction_make_divisible
This function is taken from the original tf repo. It ensures that all layers have a channel number that is divisible by 8 It can be seen
python/oneflow/test/expensive/pytorch_ghostnet.py:24
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
python/oneflow/test/expensive/pytorch_res2net.py:142
← previousnext →1,501–1,600 of 28,456, ranked by callers