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
↓ 6 callersMethodTaskNode4Id
oneflow/core/graph/task_graph_rebuild_ctx.cpp:22
↓ 6 callersFunctionThrdId4ActorId
oneflow/core/graph/task_id.cpp:77
↓ 6 callersMethodToDotWithFilePath
oneflow/core/graph/graph.h:376
↓ 6 callersMethodTopoForEachNodeWithCtrlEdge
oneflow/core/graph/op_graph.cpp:488
↓ 6 callersMethodTopoForEachNodeWithErrorCaptured
oneflow/core/graph/graph.h:267
↓ 6 callersMethodTopology
oneflow/core/hardware/node_device_descriptor.cpp:288
↓ 6 callersMethodUniqueDiffIndicesPtr
oneflow/user/kernels/model_update_kernels.cpp:52
↓ 6 callersMethodUniqueDiffValuesPtr
oneflow/user/kernels/model_update_kernels.cpp:56
↓ 6 callersMethodVisit
oneflow/maybe/variant.h:183
↓ 6 callersMethod_MakeHob
(self, other, cmp_str, cmp_func)
python/oneflow/support/high_order_bool.py:167
↓ 6 callersFunction__check
(test_case, input, dim, keepdim, device)
python/oneflow/test/modules/test_amax.py:26
↓ 6 callersFunction__check
(test_case, input, dim, keepdim, device)
python/oneflow/test/modules/test_amin.py:25
↓ 6 callersFunction__hlt
oneflow/core/device/cuda_pseudo_half.h:129
↓ 6 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytorch_poolformer.py:110
↓ 6 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
python/oneflow/test/expensive/pytorch_swin_transformer.py:22
↓ 6 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph_io_check.py:304
↓ 6 callersMethod__print
r"""Do print according to info level.
python/oneflow/nn/graph/proxy.py:576
↓ 6 callersMethod__setattr__
(self, name: str, value=None)
python/oneflow/nn/graph/proxy.py:436
↓ 6 callersFunction_broadcast_py_object
(obj, src: int = 0)
python/oneflow/framework/check_point_v2.py:289
↓ 6 callersFunction_check_equal
(test_case, lhs, rhs, rtol=1e-3, atol=1e-3)
python/oneflow/test/modules/test_global_higher_derivative_pow.py:24
↓ 6 callersFunction_check_equal
(test_case, lhs, rhs, rtol=1e-3, atol=1e-3)
python/oneflow/test/modules/test_higher_derivative_pow.py:24
↓ 6 callersFunction_get_tensor_mem
(input)
python/oneflow/test/expensive/test_tensor_offload.py:86
↓ 6 callersFunction_handle_meta_args
( input, size: Union[_size_any_t, List[int], flow.Size, None] = None, dtype: Optional[flow.dtype]
python/oneflow/nn/modules/constant.py:107
↓ 6 callersMethod_oneflow_internal_graphblock__set_origin
(self, origin)
python/oneflow/nn/graph/proxy.py:101
↓ 6 callersFunction_stack
Creates a stack of inverted residuals.
python/oneflow/test/expensive/pytorch_mnasnet.py:80
↓ 6 callersFunction_test_advanced_indexing_array
(test_case, numpy_x, dtype)
python/oneflow/test/tensor/test_tensor_indexing.py:202
↓ 6 callersFunction_test_conv2d_backward
( test_case, conv, data, weight, data_grad, weight_grad, bias=None, device="cu
python/oneflow/test/modules/test_conv2d.py:1167
↓ 6 callersFunction_test_fused_multi_head_attention_inference
( test_case, batch_size, num_heads, query_seq_len, kv_seq_len, query_head_size, va
oneflow/ir/test/OneFlow/with_cuda/test_fused_multi_head_attention_inference.py:119
↓ 6 callersMethodacc_grad
oneflow/core/framework/tensor_impl.cpp:52
↓ 6 callersMethodactor_cmd
oneflow/core/lazy/actor/actor_message.cpp:88
↓ 6 callersFunctionaligned_alloc
oneflow/user/kernels/pocketfft_hdronly.h:203
↓ 6 callersMethodcomm
oneflow/user/kernels/nccl_logical_2d_sbp_kernels.cpp:42
↓ 6 callersFunctioncompare
(a, b, a_name, b_name, atol=1e-5, rtol=1e-8)
python/oneflow/test/modules/test_rmsnorm.py:95
↓ 6 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-8)
python/oneflow/test/modules/test_fused_gelu_mul.py:43
↓ 6 callersMethodcompute_time
oneflow/core/vm/remat/disjoint_set.h:43
↓ 6 callersMethodcontaining_current_rank
oneflow/core/job/parallel_desc.h:57
↓ 6 callersMethodcreateInt32Vec
oneflow/ir/oneflow-lite/include/OneFlow/FlatbufferUtils.h:105
↓ 6 callersFunctioncuda_round_helper
oneflow/core/common/data_type_converter.h:192
↓ 6 callersMethodcurrent_grad
oneflow/core/framework/tensor_impl.cpp:54
↓ 6 callersMethoddata_type
oneflow/core/common/tensor_buffer.cpp:169
↓ 6 callersMethoddevice_type
oneflow/core/job/collective_boxing/scheduler.cpp:88
↓ 6 callersMethoddivides
oneflow/core/ep/common/primitive/broadcast_elementwise_unary.h:157
↓ 6 callersFunctiondo_pad_conv_graph
(test_case, with_cuda, with_bias, with_nchw=True)
oneflow/ir/test/OneFlow/fuse/test_fuse_pad_conv.py:28
↓ 6 callersMethoddst_placed_nd_sbp
oneflow/core/boxing/boxing_interpreter_status.h:66
↓ 6 callersMethoddst_split_axis
oneflow/user/kernels/nccl_logical_fusion_kernel.cpp:102
↓ 6 callersFunctionembedding
r"""A simple lookup table that looks up embeddings in a fixed dictionary and size. This module is often used to retrieve word embeddings using in
python/oneflow/nn/modules/sparse.py:193
↓ 6 callersFunctionerf
oneflow/core/common/bfloat16_math.h:28
↓ 6 callersMethodforward
(self)
python/oneflow/nn/modules/constant.py:70
↓ 6 callersFunctiongen_params_1d_fft
(lower_n_dims=1, upper_n_dims=5)
python/oneflow/test/modules/test_fft.py:54
↓ 6 callersFunctiongen_params_2d_fft
(lower_n_dims=2, upper_n_dims=5)
python/oneflow/test/modules/test_fft.py:80
↓ 6 callersFunctiongen_params_nd_fft
(lower_n_dims=2, upper_n_dims=5)
python/oneflow/test/modules/test_fft.py:117
↓ 6 callersFunctiongetHierarchy
oneflow/ir/lib/OneFlow/OneFlowOpTraits.cpp:112
↓ 6 callersFunctiongetRandBoolvalue
()
python/oneflow/test/modules/test_stft.py:30
↓ 6 callersFunctiongetScopeSymbolID
oneflow/ir/lib/OneFlow/OneFlowOpTraits.cpp:108
↓ 6 callersMethodget_graph
(self)
python/oneflow/framework/infer_compiler/with_oneflow_compile.py:273
↓ 6 callersMethodget_lines
oneflow/extension/stack/stacktrace.h:3657
↓ 6 callersMethodget_schema_name
(self)
tools/functional/generator.py:421
↓ 6 callersFunctionget_target_prime
oneflow/user/kernels/ctc_loss_kernel_util.cpp:21
↓ 6 callersFunctionhandler
(signum, frame)
python/oneflow/utils/data/_utils/signal_handling.py:78
↓ 6 callersMethodhas_ctrl_bootstrap_conf
oneflow/core/job/env_desc.h:34
↓ 6 callersFunctionis_complex_dtype
(dtype)
python/oneflow/test/modules/test_fft.py:39
↓ 6 callersMethodis_open
oneflow/extension/stack/stacktrace.h:3612
↓ 6 callersMethodlazy_origin_builder
(self)
python/oneflow/nn/graph/proxy.py:644
↓ 6 callersMethodload
(self)
python/oneflow/framework/model.py:560
↓ 6 callersFunctionlog2
oneflow/core/common/bfloat16_math.h:35
↓ 6 callersMethodlower
oneflow/user/kernels/model_update_kernels.cpp:86
↓ 6 callersMethodmachine
oneflow/core/job/resource_desc.cpp:37
↓ 6 callersMethodmake_shared
oneflow/core/common/single_thread_obj_pool.h:49
↓ 6 callersMethodmethod_overrided
(self, method_name: str = None)
python/oneflow/framework/model.py:250
↓ 6 callersMethodmin_register_num
min_register_num_, max_register_num_
oneflow/core/register/register_desc.h:47
↓ 6 callersMethodmut_access_list
Setters
oneflow/core/vm/vm_object.h:100
↓ 6 callersMethodmut_call_ctx
oneflow/core/vm/op_call_instruction_policy.h:94
↓ 6 callersMethodmut_eager_local_tensor_impl
Getters valid only for EagerLocalTensor
oneflow/core/framework/tensor.h:77
↓ 6 callersMethodmut_notifier
oneflow/core/vm/thread_ctx.h:49
↓ 6 callersMethodmut_response
oneflow/core/control/ctrl_call.h:48
↓ 6 callersMethodmutable_hook
oneflow/core/intrusive/skiplist_hook.h:78
↓ 6 callersMethodnode_num
oneflow/core/graph/graph.h:149
↓ 6 callersMethodorder_in_chain
oneflow/core/graph/task_node.h:60
↓ 6 callersMethodparallel_desc_sym
oneflow/core/graph/op_graph.h:45
↓ 6 callersFunctionproxy_class
(cls: type)
python/oneflow/framework/infer_compiler/transform/builtin_transform.py:66
↓ 6 callersMethodrank
oneflow/core/control/ctrl_bootstrap.h:59
↓ 6 callersMethodrecomputation_num
oneflow/core/vm/remat/env.h:55
↓ 6 callersFunctionrematable_storage
oneflow/api/python/remat/remat.cpp:30
↓ 6 callersMethodset_bar
oneflow/core/intrusive/intrusive_core_test.cpp:64
↓ 6 callersMethodset_enable_reuse_mem
oneflow/core/register/register_desc.h:68
↓ 6 callersMethodset_last_used_stream
oneflow/core/eager/eager_blob_object.cpp:146
↓ 6 callersMethodset_machine_id
oneflow/core/graph/task_node.cpp:75
↓ 6 callersFunctionskip_unless
(n, d)
python/oneflow/framework/unittest.py:188
↓ 6 callersFunctionstr2int
(env_config)
python/oneflow/framework/env_util.py:170
↓ 6 callersMethodstride_ptr
oneflow/core/eager/eager_blob_object.cpp:84
↓ 6 callersFunctiontanh
oneflow/core/common/bfloat16_math.h:46
↓ 6 callersMethodtrain
r""" train(mode=True) Sets the module in training mode. This has any effect only on certain modules. See documentati
python/oneflow/nn/modules/module.py:758
↓ 6 callersFunctiontrain_with_graph
(call_cnt=0, state_dict_file=None, last_state_dict=None)
python/oneflow/test/graph/test_graph_save_load_global_b_s.py:34
↓ 6 callersMethodupdate_parameters
(self, model)
python/oneflow/nn/optimizer/swa_utils.py:156
↓ 6 callersMethodupper
oneflow/user/kernels/model_update_kernels.cpp:87
↓ 6 callersFunctionworld_size
()
python/oneflow/framework/unittest.py:119
↓ 5 callersMethodAccessBlobByCallback
oneflow/core/framework/instructions_builder.cpp:715
↓ 5 callersMethodAddCallback
oneflow/core/lazy/actor/actor.cpp:815
↓ 5 callersMethodAlignedByteSizeOfBlobHeader
oneflow/core/register/blob_desc.cpp:101
↓ 5 callersMethodAlloc
oneflow/core/ep/cpu/cpu_device.cpp:40
← previousnext →1,101–1,200 of 28,456, ranked by callers