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
↓ 9 callersFunctionPySbpParallelCheck
SBP
oneflow/api/python/functional/common.cpp:294
↓ 9 callersFunctionRandomShape
oneflow/api/cpp/tests/api_test.cpp:33
↓ 9 callersMethodReceive
oneflow/core/common/channel.h:60
↓ 9 callersMethodRemoveOnes
oneflow/core/common/shape.cpp:180
↓ 9 callersFunctionSafeAdd3D
oneflow/user/kernels/grid_sample_kernel_util.h:309
↓ 9 callersFunctionSetDynamicLossScaleSkipIf
oneflow/core/job_rewriter/optimizer.cpp:67
↓ 9 callersFunctionTotalByteSize4BlobDesc
oneflow/core/framework/sbp_infer_util.cpp:631
↓ 9 callersMethodTotalMainByteSize4AllRegst
oneflow/core/register/runtime_register_desc.cpp:99
↓ 9 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_save_load_global_b_s.py:197
↓ 9 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_depend.py:41
↓ 9 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_free_eager_tensor.py:94
↓ 9 callersFunction_build_graph_and_test
(TestModel, in_data, test_case)
python/oneflow/test/graph/test_graph_depend.py:35
↓ 9 callersFunction_grad_preprocess
(inputs, create_graph, need_graph)
python/oneflow/autograd/functional.py:81
↓ 9 callersFunction_graph_debug
(test_case, v_level=0, ranks=None, max_py_stack_depth=2)
python/oneflow/test/graph/test_graph_debug.py:29
↓ 9 callersFunction_is_path
(path_or_buffer)
python/oneflow/framework/check_point_v2.py:109
↓ 9 callersFunction_resnet
( arch: str, block: Type[Union[BasicBlock, Bottleneck]], layers: List[int], pretrained: bool,
python/oneflow/test/expensive/pytorch_resnet.py:300
↓ 9 callersMethodclip_grad
r"""Clips gradient norm of an iterable of parameters. The norm is computed over all gradients together, as if they were concatenated into a s
python/oneflow/optim/optimizer.py:406
↓ 9 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_fused_get_intersection_area.py:34
↓ 9 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_fused_center.py:35
↓ 9 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_fused_get_convex_diagonal_squared.py:38
↓ 9 callersMethoddouble
r""" double() Casts all floating point parameters and buffers to ``double`` datatype. .. note:: This met
python/oneflow/nn/modules/module.py:1648
↓ 9 callersFunctionend_index
oneflow/user/kernels/adaptive_pool_kernel_util.h:32
↓ 9 callersMethodhash_value
oneflow/core/framework/device.h:49
↓ 9 callersMethodint
(self)
python/oneflow/nn/utils/rnn.py:150
↓ 9 callersFunctionisfinite
oneflow/core/common/bfloat16.h:255
↓ 9 callersMethodjob_name
oneflow/core/vm/lazy_job_instruction_policy.h:42
↓ 9 callersMethodload
oneflow/core/framework/tensor.cpp:193
↓ 9 callersMethodmut_tensor_meta
oneflow/core/framework/tensor.h:456
↓ 9 callersMethodnbytes
oneflow/core/common/tensor_buffer.h:89
↓ 9 callersMethodpermute_hidden
( self, hx: Tuple[Tensor, Tensor], permutation: Optional[Tensor] )
python/oneflow/nn/modules/rnn.py:659
↓ 9 callersMethodset_eviction_disabled
oneflow/core/eager/tensor_storage.h:106
↓ 9 callersMethodset_gradient_accumulation_steps
r"""Set num of steps to accumulate gradient. For example: .. code-block:: python import oneflow as flow cl
python/oneflow/nn/graph/graph_config.py:219
↓ 9 callersFunctionstart_index
oneflow/user/kernels/adaptive_pool_kernel_util.h:28
↓ 9 callersMethodto_string
(self, to_cpp=False)
tools/functional/generator.py:358
↓ 9 callersFunctiontrunc
oneflow/core/common/bfloat16_math.h:47
↓ 8 callersMethodAdd
oneflow/core/intrusive/flat_msg.h:481
↓ 8 callersMethodBind
oneflow/maybe/maybe.h:149
↓ 8 callersFunctionBuildTensor
oneflow/core/framework/op_interpreter/lazy_op_interpreter.cpp:52
↓ 8 callersFunctionCastDeviceForCPUScalarTensor
oneflow/core/functional/impl/binary_functor.cpp:55
↓ 8 callersFunctionCheckAxis
oneflow/core/functional/impl/common.cpp:149
↓ 8 callersMethodClear
oneflow/core/job/cluster_instruction.cpp:50
↓ 8 callersMethodClose
oneflow/core/framework/nn_graph.cpp:108
↓ 8 callersMethodContaining
oneflow/core/common/shape.cpp:49
↓ 8 callersFunctionCreateMatWithPtr
oneflow/user/image/image_util.h:34
↓ 8 callersMethodDestroyStream
oneflow/core/ep/cpu/cpu_device.cpp:30
↓ 8 callersFunctionErrorString4Inputs
oneflow/core/framework/op_interpreter/op_interpreter_util.cpp:50
↓ 8 callersMethodFillOpParallelDesc
oneflow/core/operator/operator.cpp:147
↓ 8 callersFunctionFlattenArray
(input_array)
python/oneflow/test/graph/test_util.py:84
↓ 8 callersMethodFlush
oneflow/core/persistence/persistent_out_stream.cpp:37
↓ 8 callersMethodForEachNodeOnValidOutEdge
oneflow/core/graph/inplace_lbi_graph.cpp:171
↓ 8 callersMethodGetComponentSbpId
oneflow/core/auto_parallel/sbp_node.cpp:855
↓ 8 callersMethodGetEagerBoxingInterpreter
oneflow/core/boxing/eager_boxing_interpreter_mgr.cpp:196
↓ 8 callersFunctionGetEnvVar
oneflow/api/cpp/env_impl.cpp:52
↓ 8 callersMethodGetParallelDescSymbolId
oneflow/core/job/scope.cpp:67
↓ 8 callersFunctionGetShape5D
oneflow/user/kernels/adaptive_pool_kernel_util.h:42
↓ 8 callersFunctionGetStreamId
oneflow/core/graph/task_graph.cpp:297
↓ 8 callersFunctionGetTensorDevice4CurrentProcessCtx
oneflow/core/job/parallel_desc.cpp:170
↓ 8 callersFunctionHasValue
oneflow/maybe/optional.h:335
↓ 8 callersMethodInit
oneflow/core/framework/device.cpp:50
↓ 8 callersFunctionIsComplexDataType
oneflow/core/common/data_type.cpp:59
↓ 8 callersFunctionIsCpuOnly
oneflow/core/operator/operator.cpp:1295
↓ 8 callersFunctionIsForkedSubProcess
oneflow/core/platform/lib/pthread_fork.cpp:28
↓ 8 callersMethodIsHostMemoryInput
oneflow/core/framework/op_expr.h:158
↓ 8 callersMethodIsLastIRPass
oneflow/ir/oneflow-extension/ir_pass.cpp:65
↓ 8 callersFunctionIsNdSbpBoxingAcyclic
oneflow/core/framework/placement_sbp_util.cpp:416
↓ 8 callersFunctionMakeParallelConf
oneflow/core/framework/parallel_conf_util.cpp:34
↓ 8 callersFunctionMakeScopeSymbolId
oneflow/core/job_rewriter/autograd.cpp:270
↓ 8 callersMethodMemZoneId121
oneflow/core/graph/task_node.cpp:271
↓ 8 callersMethodMoveTo
oneflow/core/intrusive/list.h:127
↓ 8 callersFunctionNewMatmulPrimitive
oneflow/user/kernels/conv_kernels.cpp:30
↓ 8 callersMethodNotify
oneflow/core/common/notifier.cpp:22
↓ 8 callersFunctionOpNodeHasScope
oneflow/core/job_rewriter/pipeline_buffer_pass.cpp:72
↓ 8 callersFunctionPMINPLACE
oneflow/user/kernels/pocketfft_hdronly.h:345
↓ 8 callersFunctionParseDims
oneflow/core/functional/impl/fused_attention_functor.cpp:39
↓ 8 callersMethodProcess
oneflow/core/control/ctrl_call.h:56
↓ 8 callersMethodPush
oneflow/core/common/buffer.h:48
↓ 8 callersMethodSend
oneflow/core/common/channel.h:47
↓ 8 callersMethodSerialize
oneflow/core/hardware/node_device_descriptor.cpp:292
↓ 8 callersMethodSetAsActiveDevice
oneflow/core/ep/cpu/cpu_device.cpp:26
↓ 8 callersFunctionSplit
oneflow/core/common/str_util.cpp:64
↓ 8 callersFunctionTEST
oneflow/core/framework/placement_sbp_util_test.cpp:164
↓ 8 callersFunctionTEST
oneflow/core/ndarray/cpu_slice_var_ndarray_test.cpp:23
↓ 8 callersMethodTotalMachineNum
oneflow/core/job/env_desc.cpp:31
↓ 8 callersMethodTryInit
(self)
python/oneflow/framework/multi_client_session.py:53
↓ 8 callersMethod__Init__
oneflow/core/intrusive/base.h:24
↓ 8 callersMethod__class__
(self)
python/oneflow/nn/graph/proxy.py:221
↓ 8 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytorch_uniformer.py:27
↓ 8 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytroch_mlp_mixer.py:29
↓ 8 callersMethod_generate_grad_clip_conf_for_optim_conf
(self, param_group, optimizer_conf)
python/oneflow/optim/optimizer.py:492
↓ 8 callersMethod_generate_lr_scale_for_optim_conf
(self, param_group, optimizer_conf)
python/oneflow/optim/optimizer.py:504
↓ 8 callersFunction_randint
Get a random integer in the range [low, high).
python/oneflow/test/tensor/test_global_tensor_indexing.py:28
↓ 8 callersMethod_state
(self)
python/oneflow/nn/graph/graph.py:701
↓ 8 callersFunction_test_bce_loss
(dim=int, with_logits: bool = False)
python/oneflow/test/modules/test_loss.py:223
↓ 8 callersFunction_test_cross_entropy_loss
(dim: int, prob: bool = False)
python/oneflow/test/modules/test_loss.py:99
↓ 8 callersFunction_test_image_resize_with_cv
( test_case, image_files, target_size, min_size=None, max_size=None, keep_aspect_ratio
python/oneflow/test/modules/test_image_resize.py:138
↓ 8 callersFunction_test_linear_graph_train_with_lr_sch
( test_case, iter_num, device, get_opt_and_lr_sch )
python/oneflow/test/graph/test_graph_lrs.py:27
↓ 8 callersFunction_test_nll_loss
( test_case, has_weight=False, split_batch_dim=False, split_class_dim=False )
python/oneflow/test/modules/test_nll_loss.py:27
↓ 8 callersFunctionare_tensors_equal
(a, b)
python/oneflow/test/modules/test_dlpack.py:39
↓ 8 callersMethodavailable_regst_desc_cnt
oneflow/core/lazy/actor/register_slot.h:31
↓ 8 callersMethodbackward
(ctx, z_grad)
python/oneflow/test/modules/test_autograd_function.py:194
← previousnext →801–900 of 28,456, ranked by callers