MCPcopy Create free account

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

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

↓ 8 callersMethodbuffer
oneflow/core/vm/instruction.h:43
↓ 8 callersFunctionclip_grad_norm_np
(np_grad, max_norm, norm_type)
python/oneflow/test/modules/optimizer_test_util.py:19
↓ 8 callersFunctionclip_grad_norm_np
(np_grad, max_norm, norm_type)
python/oneflow/test/graph/optimizer_test_util.py:19
↓ 8 callersFunctioncompare
(a, b, a_name, b_name, atol=1e-5, rtol=1e-8)
python/oneflow/test/modules/test_layer_norm.py:120
↓ 8 callersFunctioncompare
(a, b, a_name, b_name, atol=1e-5, rtol=1e-8)
python/oneflow/test/modules/test_fused_bias_add_scale_mask_softmax_dropout.py:94
↓ 8 callersFunctionconvertToSignless
oneflow/ir/lib/OneFlow/Conversion/OneFlowToTosa.cpp:57
↓ 8 callersFunctiondo_test_softmax
(batch_size: int, log_softmax: bool = False)
python/oneflow/test/modules/test_activation.py:456
↓ 8 callersMethodenable_auto_mixed_precision
oneflow/core/job/job_desc.h:55
↓ 8 callersMethodexec
oneflow/user/kernels/pocketfft_hdronly.h:1467
↓ 8 callersFunctionf
(x, requires_grad)
python/oneflow/test/modules/test_loss_global.py:64
↓ 8 callersFunctiongetDeviceName
oneflow/ir/lib/OneFlow/OneFlowOpTraits.cpp:103
↓ 8 callersMethodgetDeviceTag
oneflow/ir/include/OneFlow/OneFlowOpTraits.h:61
↓ 8 callersFunctiongetSI64IntegerAttr
oneflow/ir/lib/OneFlow/OneFlowRewrites.cpp:131
↓ 8 callersMethodget_args
oneflow/ir/oneflow-extension/include/PyAst/Ast.h:352
↓ 8 callersFunctionhash
(value)
python/oneflow/utils/data/_utils/worker.py:198
↓ 8 callersMethodid
oneflow/core/eager/tensor_storage.h:108
↓ 8 callersMethodis_in_memory
oneflow/core/eager/tensor_storage.h:102
↓ 8 callersMethodis_pinned
oneflow/core/framework/tensor_impl.cpp:144
↓ 8 callersFunctionisinf
oneflow/core/common/bfloat16.h:253
↓ 8 callersMethodmut_data_regst_time_shape
oneflow/core/register/register_desc.h:92
↓ 8 callersMethodmut_grad_fn_node
oneflow/core/autograd/autograd_captured_tensor.h:45
↓ 8 callersMethodnamed_buffers
r""" named_buffers(prefix="", recurse=True) -> Iterator[Tuple[str, Tensor]] Returns an iterator over module buffers, yielding
python/oneflow/nn/modules/module.py:606
↓ 8 callersMethodnamed_modules
r""" named_modules(memo=None, prefix="") Returns an iterator over all modules in the network, yielding both the name
python/oneflow/nn/modules/module.py:714
↓ 8 callersMethodobn4lbi
oneflow/core/operator/operator.cpp:103
↓ 8 callersFunctionrsqrt
oneflow/core/common/bfloat16_math.h:50
↓ 8 callersMethodstream
oneflow/core/lazy/actor/light_actor.cpp:662
↓ 8 callersMethodstream_type
oneflow/core/vm/stream.h:63
↓ 8 callersMethodthread_num
oneflow/core/thread/thread_pool.h:31
↓ 7 callersMethodBindBnWithOneOfTheRegsts
oneflow/core/graph/exec_graph.cpp:51
↓ 7 callersFunctionCheckInplaceShapeCanExpandTo
oneflow/core/functional/impl/common.cpp:180
↓ 7 callersFunctionCheckScalarDataType
oneflow/user/ops/model_update_ops.cpp:45
↓ 7 callersFunctionCheckScalarShape
oneflow/user/ops/multi_tensor_model_update_ops.cpp:40
↓ 7 callersFunctionCheckScalarShape
oneflow/user/ops/model_update_ops.cpp:39
↓ 7 callersMethodConnectToSrcNodeWithSlice
oneflow/core/graph/slice_boxing_task_node.cpp:80
↓ 7 callersMethodCopy
oneflow/core/register/tensor_slice_copier.cpp:56
↓ 7 callersMethodCreateStream
oneflow/core/ep/cpu/cpu_device.cpp:28
↓ 7 callersMethodDebugStr
oneflow/core/common/high_order_bool.h:80
↓ 7 callersFunctionExpandDimIf0D
oneflow/core/common/shape.cpp:123
↓ 7 callersMethodForEachOutDataEdge
oneflow/core/graph/task_node.cpp:490
↓ 7 callersFunctionGenCvMat4ImageBuffer
oneflow/user/image/image_util.cpp:41
↓ 7 callersFunctionGenTensorDescFromBlobDesc
oneflow/core/operator/user_op.cpp:47
↓ 7 callersFunctionGet3DVec
oneflow/user/utils/pool_util.h:63
↓ 7 callersFunctionGetCallbackNotifierBufferName
oneflow/core/common/buffer_manager.h:58
↓ 7 callersMethodGetInputBlobFastestTimeShape
oneflow/core/operator/operator.cpp:491
↓ 7 callersMethodGetNextRankInRing
oneflow/core/job/rank_group.cpp:106
↓ 7 callersFunctionGetSelectedParallelIds
oneflow/core/framework/placement_sbp_util.cpp:142
↓ 7 callersFunctionGetUserOpCommonAttrs
oneflow/ir/lib/OneFlow/OneFlowRewrites.cpp:142
↓ 7 callersFunctionINTRUSIVE_UNSAFE_FOR_EACH_PTR
oneflow/core/vm/virtual_machine_engine.cpp:141
↓ 7 callersFunctionImageObjectGetSbp
oneflow/user/ops/image_object_preprocess_ops.cpp:23
↓ 7 callersMethodIndex
oneflow/maybe/variant.h:227
↓ 7 callersFunctionInferUnifiedShapeForBroadcasting
oneflow/core/functional/impl/common.cpp:36
↓ 7 callersMethodInterpret
oneflow/core/boxing/eager_boxing_interpreter.cpp:35
↓ 7 callersMethodIsBool
oneflow/core/common/scalar.h:88
↓ 7 callersFunctionIsDependyOp
oneflow/core/job_rewriter/prune_depend_op_pass.cpp:39
↓ 7 callersMethodIsInMutOpTransaction
oneflow/core/job/job_builder.cpp:342
↓ 7 callersMethodMakePredicatorIsOpNameDataOrCtrlReachable
oneflow/core/graph/op_graph.cpp:500
↓ 7 callersMethodMutOpTransactionGet
oneflow/core/job/job_builder.cpp:347
↓ 7 callersMethodMutOpTransactionMut
oneflow/core/job/job_builder.cpp:351
↓ 7 callersMethodMutRequestEntry
oneflow/core/job/collective_boxing/request_store.h:96
↓ 7 callersMethodOffsetToNdIndex
oneflow/core/ep/common/primitive/constant_pad.h:103
↓ 7 callersFunctionOptionalBoxing
oneflow/core/boxing/eager_boxing_interpreter.cpp:218
↓ 7 callersFunctionPyDTypeInfo_UnpackDType
oneflow/api/python/framework/typeinfo.h:39
↓ 7 callersFunctionPySequenceCheck
oneflow/api/python/functional/common.cpp:89
↓ 7 callersFunctionRecv
oneflow/user/kernels/communicate_util.cpp:52
↓ 7 callersFunctionRelu
oneflow/api/cpp/tests/nn_test.cpp:29
↓ 7 callersFunctionReplaceSlashToDash4Lbn
oneflow/core/job_rewriter/pass_util.cpp:26
↓ 7 callersFunctionRun
oneflow/core/vm/vm_util.cpp:31
↓ 7 callersFunctionSend
oneflow/user/kernels/communicate_util.cpp:38
↓ 7 callersFunctionSetNdSbpDim
Let a nd sbp be consistent with the given hierarchy number
oneflow/core/auto_parallel/boxing_collector.cpp:57
↓ 7 callersMethodSyncAccessBlobByCallback
oneflow/core/framework/instructions_builder.cpp:629
↓ 7 callersFunctionThrowError
oneflow/core/common/error.cpp:376
↓ 7 callersMethodTotalSeparatedHeaderByteSize4AllRegst
oneflow/core/register/runtime_register_desc.cpp:119
↓ 7 callersMethod__init__
(self, file_like)
python/oneflow/framework/check_point_v2.py:68
↓ 7 callersMethod__init__
( self, cnf: MBConvConfig, stochastic_depth_prob: float, norm_layer: Callable[
python/oneflow/test/expensive/pytorch_efficientnet.py:254
↓ 7 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytorch_crossformer.py:25
↓ 7 callersMethod__init__
( self, img_size=224, patch_size=16, in_chans=3, num_classes=1000,
python/oneflow/test/expensive/pytorch_levit.py:377
↓ 7 callersMethod__init__
( self, num_classes: int = 1000, aux_logits: bool = True, transform_input: boo
python/oneflow/test/expensive/pytorch_inception_v3.py:27
↓ 7 callersMethod__init__
(self, *args, **kwargs)
python/oneflow/nn/modules/rnn.py:351
↓ 7 callersMethod__init__
(self, padding: Union[int, tuple, list])
python/oneflow/nn/modules/padding.py:541
↓ 7 callersMethod__init__
An ecution proxy of nn.Module or Tensor. A proxy contains the original data(nn.Module or Tensor) and a graph representation of the original
python/oneflow/nn/graph/proxy.py:61
↓ 7 callersFunction_blob_desc_repr
(blob_desc)
python/oneflow/nn/graph/util.py:49
↓ 7 callersFunction_fused_mha
( query, key, value, num_heads, attn_mask_type="none", attn_bias=None, causal_diag
python/oneflow/test/modules/test_fused_attention_ops.py:157
↓ 7 callersFunction_nd_sbp2repr
(nd_sbp)
python/oneflow/nn/graph/util.py:32
↓ 7 callersFunction_ref
( query, key, value, num_heads, attn_mask_type="none", attn_bias=None, causal_diag
python/oneflow/test/modules/test_fused_attention_ops.py:27
↓ 7 callersFunction_test_conv2d
( test_case, conv, data, weight, output, bias=None, device="cuda", )
python/oneflow/test/modules/test_conv2d.py:1154
↓ 7 callersFunctionassert_set_eq
(tensor, indexer, val)
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:454
↓ 7 callersFunctionassert_set_eq
(tensor, indexer, val)
python/oneflow/test/tensor/test_tensor_indexing2.py:495
↓ 7 callersFunctionassert_set_eq
(tensor, indexer, val)
python/oneflow/test/tensor/test_global_tensor_indexing.py:465
↓ 7 callersMethodboxing_interpreter_status
oneflow/core/boxing/eager_boxing_interpreter.h:75
↓ 7 callersMethodclose
(self)
python/oneflow/hub.py:71
↓ 7 callersMethodcomm
oneflow/user/kernels/nccl_logical_kernels.cpp:41
↓ 7 callersMethodcompute_op_type_name
oneflow/core/eager/tensor_storage.cpp:154
↓ 7 callersMethodconsumer_nd_sbp_constraint
Getters/Setters valid only for EagerGlobalTensor
oneflow/core/framework/tensor.h:87
↓ 7 callersMethodcreateOffsetVecDestructive
Creates an offset vector with the given values. Unlike createOffsetVec this will destroy the input values array during serialization but be much faste
oneflow/ir/oneflow-lite/include/OneFlow/FlatbufferUtils.h:98
↓ 7 callersMethoddebug_str
(self, ctx, display_result=True)
python/oneflow/support/high_order_bool.py:35
↓ 7 callersMethoddevice_index
oneflow/core/framework/auto_random_generator.h:39
↓ 7 callersMethoddevice_type
oneflow/core/ep/cpu/cpu_stream.cpp:23
↓ 7 callersMethodfold
oneflow/ir/lib/OneFlow/OneFlowOpFolders.cpp:119
↓ 7 callersFunctionget_tensor_shape
(call_pytorch)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:115
↓ 7 callersMethodhas_cuda_time
(self)
python/oneflow/profiler/events.py:84
← previousnext →901–1,000 of 28,456, ranked by callers