MCPcopy Create free account

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

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

↓ 11 callersMethodDebugName
oneflow/core/vm/instruction.cpp:30
↓ 11 callersFunctionDeviceType4DeviceTag
oneflow/core/framework/to_string.cpp:32
↓ 11 callersFunctionGetOpKernelRandomSeedInCurrentRank
oneflow/user/kernels/random_seed_util.cpp:63
↓ 11 callersFunctionIsStrInt
oneflow/core/common/str_util.h:25
↓ 11 callersFunctionMakeGenerator
oneflow/core/framework/random_generator.cpp:196
↓ 11 callersMethodMap
oneflow/maybe/maybe.h:139
↓ 11 callersFunctionNumAxes
oneflow/core/common/flat_shape.h:40
↓ 11 callersMethodNumAxes
oneflow/core/register/tensor_slice_view.cpp:98
↓ 11 callersMethodNumOfLbi
oneflow/core/register/register_desc.h:62
↓ 11 callersMethodPopFrontRegsts
oneflow/core/lazy/actor/register_slot.cpp:60
↓ 11 callersMethodSize
oneflow/user/data/coco_dataset.cpp:40
↓ 11 callersMethodSize
oneflow/core/embedding/posix_file.h:73
↓ 11 callersFunctionWithinBounds2D
oneflow/user/kernels/grid_sample_kernel_util.h:272
↓ 11 callersMethod__init__
(self, *args, **kwargs)
python/oneflow/framework/model.py:193
↓ 11 callersFunction_autograd_grad
( outputs, inputs, grad_outputs=None, create_graph=False, retain_graph=None, )
python/oneflow/autograd/functional.py:175
↓ 11 callersFunction_check_requires_grad
(inputs, input_type, strict)
python/oneflow/autograd/functional.py:132
↓ 11 callersMethod_compile
(self, *args, **kwargs)
python/oneflow/nn/graph/graph.py:846
↓ 11 callersFunction_tuple_postprocess
(res, to_unpack)
python/oneflow/autograd/functional.py:63
↓ 11 callersMethodbody_mem_ptr
oneflow/core/register/register.h:56
↓ 11 callersMethodbuffers
r""" buffers(recurse=True) -> Iterator[Tensor] Returns an iterator over module buffers. Args: recurse (b
python/oneflow/nn/modules/module.py:582
↓ 11 callersFunctionclosure
()
python/oneflow/test/modules/test_optim_lbfgs.py:213
↓ 11 callersMethoddevice_id
oneflow/api/cpp/framework/device.cpp:36
↓ 11 callersMethodeval
r""" eval() Sets the module in evaluation mode. This has any effect only on certain modules. See documentations of
python/oneflow/nn/modules/module.py:781
↓ 11 callersMethodhost
oneflow/core/control/ctrl_bootstrap.h:62
↓ 11 callersMethodis_initialized
Getters and Setters
oneflow/core/common/shape.h:136
↓ 11 callersMethodkey_averages
(self, group_by_input_shape=False, group_by_attributes=False)
python/oneflow/profiler/events.py:285
↓ 11 callersMethodmsg_type
oneflow/core/lazy/actor/actor_message.h:47
↓ 11 callersFunctionparse
(x)
python/oneflow/nn/modules/utils.py:25
↓ 11 callersMethodpin_memory
(self)
python/oneflow/nn/utils/rnn.py:113
↓ 11 callersMethodrecv
(self)
python/oneflow/multiprocessing/queue.py:34
↓ 11 callersMethodsend
(self, obj)
python/oneflow/multiprocessing/queue.py:29
↓ 10 callersFunctionBackward
oneflow/api/python/autograd/autograd.cpp:105
↓ 10 callersFunctionCheckInplaceCastValid
oneflow/core/functional/impl/common.cpp:172
↓ 10 callersFunctionCudnnSPOnePtr
oneflow/core/device/cudnn_util.cpp:222
↓ 10 callersFunctionDecideOrder
oneflow/core/auto_parallel/algorithm_util.h:60
↓ 10 callersMethodDeviceCount
oneflow/core/hardware/basic_device_descriptor_list.cpp:31
↓ 10 callersMethodEquals
oneflow/core/job/parallel_desc.cpp:214
↓ 10 callersMethodForEachNodeOnInEdge
oneflow/core/graph/node.h:95
↓ 10 callersFunctionGenArgDict
(arg_dict)
python/oneflow/test/expensive/test_util.py:45
↓ 10 callersFunctionGenUnRepeatedBn
oneflow/core/operator/operator.cpp:1291
↓ 10 callersFunctionGetAreaPixel
oneflow/user/kernels/upsample_kernel.h:49
↓ 10 callersFunctionGetBlasTransposeType
oneflow/user/kernels/matmul_kernels.cpp:29
↓ 10 callersFunctionGetDualObject
(name, pytorch, oneflow)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:783
↓ 10 callersFunctionGetGlobalParallelDescFromDevice
oneflow/core/job/global_mode.cpp:23
↓ 10 callersFunctionGetOptimizerWeightDecayRate
oneflow/core/job_rewriter/optimizer.cpp:36
↓ 10 callersMethodGetParallelDesc4BnInOp
oneflow/core/operator/operator.cpp:123
↓ 10 callersFunctionGetTensorSliceView
oneflow/core/job/nd_sbp_util.cpp:23
↓ 10 callersFunctionIsSplitSbp
oneflow/core/functional/impl/comm_functor.cpp:89
↓ 10 callersFunctionNdSbpAllSameSplitParallel
oneflow/core/job/sbp_parallel.cpp:294
↓ 10 callersFunctionNdSbpHasPartialParallel
oneflow/core/job/nd_sbp_util.cpp:125
↓ 10 callersMethodNewBuffer
oneflow/core/common/buffer_manager.h:30
↓ 10 callersFunctionParseSplitAxis
oneflow/user/ops/fused_attention_ops.cpp:197
↓ 10 callersFunctionPyArrayCheckLongScalar
oneflow/extension/python/numpy.cpp:101
↓ 10 callersMethodResetBodyMemPtr
oneflow/core/register/register.cpp:51
↓ 10 callersMethodSet
oneflow/core/framework/variable_tensor_mgr.cpp:27
↓ 10 callersMethodSoleNode
oneflow/core/graph/graph.h:307
↓ 10 callersMethodTensor4ArgNameAndIndex
oneflow/ir/lib/OneFlow/OKL/Kernel/ComputeContext.cpp:80
↓ 10 callersMethodToShape
oneflow/core/common/flat_shape.cpp:57
↓ 10 callersMethodToString
oneflow/core/framework/device.cpp:109
↓ 10 callersMethodTryPopFrontRegst
oneflow/core/lazy/actor/register_slot.cpp:50
↓ 10 callersMethod__init__
(self, module, optimizer=None)
python/oneflow/test/graph/test_to_global.py:180
↓ 10 callersMethod_apply
(self, fn)
python/oneflow/nn/modules/module.py:1270
↓ 10 callersFunction_grad_postprocess
(inputs, create_graph)
python/oneflow/autograd/functional.py:99
↓ 10 callersFunction_handle_size_arg
(size)
python/oneflow/nn/modules/utils.py:49
↓ 10 callersFunction_test_fused_multi_head_attention_inference_with_attn_bias
( test_case, batch_size, num_heads, query_seq_len, kv_seq_len, query_head_size, va
python/oneflow/test/modules/test_fused_attention_ops.py:357
↓ 10 callersFunction_test_fused_weighted_sum
(test_case, shape, n, device, dtype)
python/oneflow/test/modules/test_fused_weighted_sum.py:50
↓ 10 callersFunction_test_linear_graph_train_with_lr_sch
( test_case, iter_num, device, get_opt_and_lr_sch )
python/oneflow/test/graph/test_graph_lr_with_warmup.py:26
↓ 10 callersFunctionadd_indent
(in_s, num_spaces)
python/oneflow/nn/graph/util.py:289
↓ 10 callersMethodcudnn_handle
oneflow/core/ep/cuda/cuda_stream.cpp:293
↓ 10 callersMethoddtype
oneflow/core/common/tensor_meta.h:48
↓ 10 callersMethodenable_amp
r"""If set to true, then graph will use mixed precision mode, it means use both float16 and float32 during model training. For example:
python/oneflow/nn/graph/graph_config.py:50
↓ 10 callersMethodepsilon
oneflow/core/common/bfloat16.h:292
↓ 10 callersMethodfeatures
(self, x)
python/oneflow/test/expensive/pytorch_senet.py:330
↓ 10 callersFunctiongetDeviceTag
oneflow/ir/lib/OneFlow/OneFlowOpTraits.cpp:99
↓ 10 callersMethodis_integer
oneflow/core/framework/dtype.cpp:150
↓ 10 callersMethodis_leaf
oneflow/core/framework/tensor.h:498
↓ 10 callersMethoditer_nodes
r""" return a generator of the args tree nodes in the DFS manner. The node returned can be of type NamedArg or non-NamedArg depending
python/oneflow/framework/args_tree.py:146
↓ 10 callersMethodlogsumexp
(t)
python/oneflow/distributions/categorical.py:85
↓ 10 callersMethodmut_key
oneflow/core/intrusive/skiplist_hook.h:108
↓ 10 callersMethodset_device_type
oneflow/core/job/parallel_desc.cpp:284
↓ 10 callersMethodset_mem_block_id
oneflow/core/register/register_desc.h:83
↓ 10 callersMethodset_memory_format
oneflow/core/register/blob_desc.h:72
↓ 10 callersMethodset_state
oneflow/core/lazy/actor/light_actor.cpp:688
↓ 10 callersMethodstream_ctx
oneflow/core/lazy/actor/generic_actor_context.cpp:29
↓ 10 callersMethodunique_stream_id
oneflow/core/framework/stream.h:50
↓ 10 callersMethodupdate
(self, modules: Mapping[str, T])
python/oneflow/nn/utils/container.py:251
↓ 10 callersMethodvalue
(self)
python/oneflow/support/box.py:27
↓ 9 callersMethodAddAndInferGlobalOp
oneflow/core/job/job_build_and_infer_ctx.cpp:509
↓ 9 callersMethodAddWork
oneflow/core/thread/thread_pool.cpp:40
↓ 9 callersMethodAppend
oneflow/core/persistence/posix/posix_file_system.cpp:77
↓ 9 callersFunctionCclInitCollectiveNode
oneflow/core/graph/boxing/ccl_sub_task_graph_builder.cpp:31
↓ 9 callersFunctionClusterSync
oneflow/core/vm/vm_util.cpp:37
↓ 9 callersFunctionGetElementCount
oneflow/core/ep/common/primitive/util.h:28
↓ 9 callersFunctionInOutParallelDimReduce
oneflow/core/framework/sbp_infer_util.cpp:557
↓ 9 callersMethodInit
oneflow/core/framework/op_expr.cpp:554
↓ 9 callersMethodInitEagerBlobObject
oneflow/core/framework/tensor_impl.cpp:120
↓ 9 callersMethodIsMutRef
oneflow/core/graph/inplace_lbi_graph.cpp:179
↓ 9 callersMethodIsNCHW
oneflow/ir/lib/OneFlow/Transform/AutoNHWCOps.cpp:23
↓ 9 callersFunctionNdSbpIsAllBroadcast
oneflow/core/job/nd_sbp_util.cpp:141
↓ 9 callersFunctionParseDims
oneflow/user/ops/fused_attention_ops.cpp:23
← previousnext →701–800 of 28,456, ranked by callers