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
↓ 18 callersFunctionautotest
( n=20, auto_backward: Union[bool, str] = True, rtol=0.0001, atol=1e-05, check_graph=True,
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:1222
↓ 18 callersMethodbegin
oneflow/extension/stack/stacktrace.h:989
↓ 18 callersFunctionceil
oneflow/core/common/bfloat16_math.h:36
↓ 18 callersMethodclear
Remove all items from the ModuleDict.
python/oneflow/nn/utils/container.py:221
↓ 18 callersFunctionis_cufft_available
()
python/oneflow/test/modules/test_fft.py:31
↓ 18 callersMethodis_enabled
(self)
python/oneflow/utils/global_view/global_mode.py:106
↓ 18 callersMethodmul
oneflow/core/ep/include/primitive/fast_integer_math.h:55
↓ 18 callersFunctionnearbyint
oneflow/core/common/bfloat16_math.h:39
↓ 18 callersFunctionpack
(x)
python/oneflow/test_utils/automated_test_util/generators.py:73
↓ 18 callersMethodseed
oneflow/core/framework/random_generator.cpp:82
↓ 18 callersMethodset_stage
r"""Set stage id and placement of nn.Module in pipeline parallelism. Args: stage_id (int): stage id of this module. p
python/oneflow/nn/graph/graph_block.py:119
↓ 18 callersMethodset_value
(self, val)
python/oneflow/support/box.py:35
↓ 18 callersFunctionsetitem_and_return
(ref, idx, value)
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:42
↓ 18 callersMethodstream
oneflow/core/vm/ep_stream_policy_base.h:43
↓ 18 callersMethodzeros_
oneflow/api/cpp/framework/tensor.cpp:69
↓ 17 callersMethodConnectWithLbi
oneflow/core/graph/task_graph.cpp:1022
↓ 17 callersFunctionConstructOp
oneflow/core/operator/operator.cpp:1311
↓ 17 callersFunctionGetThreadLocalAutogradEngine
oneflow/core/autograd/autograd_engine.cpp:515
↓ 17 callersMethodLast
oneflow/core/intrusive/list.h:65
↓ 17 callersFunctionPyStringAsString
oneflow/api/python/functional/common.cpp:116
↓ 17 callersFunctionSetInputArgModifierMutable
oneflow/user/ops/multi_tensor_model_update_ops.cpp:68
↓ 17 callersMethodSoleOutEdge
oneflow/core/graph/node.h:87
↓ 17 callersMethodValue
oneflow/maybe/optional.h:215
↓ 17 callersFunction_as_tuple
(inp, arg_name=None, fn_name=None)
python/oneflow/autograd/functional.py:36
↓ 17 callersFunction_set_resource_attr
r""" set the attribute of config_proto.resource to attr_value. the attribute is specified as a string or a list of string. for example, i
python/oneflow/framework/config_util.py:27
↓ 17 callersFunctionconvert_to_tensor_tuple
(args: Optional[Union[Tensor, Sequence[Tensor]]])
python/oneflow/framework/tensor_tuple_util.py:23
↓ 17 callersFunctiondo_test_train_loss_oneflow_pytorch
( test_case, model_path: str, module_name: str, device: str = "cuda", batch_size: int = 16
python/oneflow/test_utils/oneflow_pytorch_compatibility/oneflow_pytorch_compatiblity_test.py:219
↓ 17 callersMethodis_offloaded
oneflow/core/framework/tensor.h:473
↓ 17 callersMethoditems
Return an iterable of the ModuleDict key/value pairs.
python/oneflow/nn/utils/container.py:241
↓ 17 callersFunctionlinear
r""" Applies a linear transformation to the incoming data: :math:`y = xA^T + b`. Shape: - Input: :math:`(N, *, in\_features)` N is t
python/oneflow/nn/modules/linear.py:144
↓ 17 callersMethodmap_leaf
r""" Map the leaf of the arguments into map_function(leaf).
python/oneflow/framework/args_tree.py:264
↓ 17 callersMethodmutable_job_parallel_view_conf
oneflow/core/job/job_builder.h:42
↓ 17 callersMethodshape_ptr
oneflow/core/eager/eager_blob_object.cpp:77
↓ 16 callersMethodAppendTo
oneflow/core/intrusive/list_hook.h:40
↓ 16 callersMethodAt
oneflow/api/cpp/framework/shape.cpp:51
↓ 16 callersMethodCopy
oneflow/maybe/variant.h:407
↓ 16 callersFunctionCreateLeftExtendedShape
oneflow/core/common/shape.cpp:113
↓ 16 callersFunctionGetAreaPixelScale
oneflow/user/kernels/upsample_kernel.h:36
↓ 16 callersMethodInsertRegstDescId
oneflow/core/lazy/actor/register_slot.cpp:65
↓ 16 callersMethodPushFront
oneflow/core/intrusive/list.h:93
↓ 16 callersFunctionReset
oneflow/maybe/optional.h:472
↓ 16 callersFunctionSyncAutoMemcpy
oneflow/core/kernel/kernel_util.cpp:57
↓ 16 callersMethod__init__
( self, batch_size: int = 1, random_seed: Optional[int] = None, probability: f
python/oneflow/nn/modules/dataset.py:199
↓ 16 callersFunction_normalize
(fmt)
tools/functional/generator.py:190
↓ 16 callersFunction_test_fused_multi_head_attention_inference
( 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:301
↓ 16 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-8)
python/oneflow/test/modules/test_fused_get_boundding_boxes_coord.py:87
↓ 16 callersMethodcreateString
Creates a string with the given string contents (including zeros).
oneflow/ir/oneflow-lite/include/OneFlow/FlatbufferUtils.h:73
↓ 16 callersMethoddevice_index
oneflow/core/graph/stream_id.h:44
↓ 16 callersFunctionevict
(tensor)
python/oneflow/test/expensive/_test_remat.py:33
↓ 16 callersMethodexecute_sql
(self, sql)
python/oneflow/utils/insight/sqlite_to_google_trace_event.py:38
↓ 16 callersFunctionfmod
oneflow/core/common/bfloat16_math.h:60
↓ 16 callersMethodinput_arg_tuple
oneflow/user/kernels/stateful_opkernel.h:88
↓ 16 callersMethodmem_block_offset
oneflow/core/register/register_desc.cpp:37
↓ 16 callersMethodmut_request
oneflow/core/control/rpc_client.cpp:45
↓ 16 callersMethodnum_ranks
oneflow/user/kernels/nccl_logical_fusion_kernel.cpp:91
↓ 16 callersMethodprod
oneflow/user/kernels/pocketfft_hdronly.h:525
↓ 16 callersMethodset_current_seed
oneflow/core/framework/random_generator.cpp:66
↓ 16 callersMethodsize
oneflow/extension/python/numpy.h:33
↓ 16 callersMethodto
(self, *args, **kwargs)
python/oneflow/framework/infer_compiler/with_oneflow_compile.py:57
↓ 15 callersMethodAddNode
oneflow/core/autograd/autograd_engine.cpp:490
↓ 15 callersFunctionBuildSubTskGphBuilderStatus
oneflow/core/graph/boxing/sub_task_graph_builder_status_util.cpp:20
↓ 15 callersFunctionCheckShapeLike
oneflow/user/ops/model_update_ops.cpp:26
↓ 15 callersFunctionCreateReducedShape
oneflow/core/common/shape.cpp:104
↓ 15 callersFunctionFlatten
oneflow/core/framework/placement_sbp_util.cpp:258
↓ 15 callersMethodGetProxyNode
oneflow/core/graph/task_graph.cpp:545
↓ 15 callersMethodGetTaskType
Others
oneflow/core/graph/task_node.h:99
↓ 15 callersMethodInit
oneflow/core/graph/op_graph.cpp:189
↓ 15 callersFunctionIsNodeInList
oneflow/core/job_rewriter/pass_util.cpp:20
↓ 15 callersFunctionLoadGraph
oneflow/api/cpp/tests/graph_test.cpp:35
↓ 15 callersMethodLogicalTensorDesc4ArgNameAndIndex
oneflow/user/kernels/stateful_opkernel.cpp:357
↓ 15 callersMethodMaybeForEachNode
oneflow/core/graph/graph.h:205
↓ 15 callersMethodMutOpOnlyOnce
oneflow/core/job/job_builder.cpp:325
↓ 15 callersMethodNdSbp4BnInOp
oneflow/core/operator/operator.cpp:1078
↓ 15 callersMethodParallelFor
oneflow/core/ep/cpu/cpu_stream.h:80
↓ 15 callersMethodPullKV
oneflow/core/rpc/lib/local.cpp:108
↓ 15 callersMethodPushKV
oneflow/core/rpc/lib/local.cpp:83
↓ 15 callersMethodSendResponse
oneflow/core/control/ctrl_call.h:69
↓ 15 callersMethodSoleInEdge
oneflow/core/graph/node.h:83
↓ 15 callersFunction_test_rmsnorm
( test_case, shape, normalized_shape, affine=True, eps=1e-6, dtype=flow.float32, d
python/oneflow/test/modules/test_rmsnorm.py:45
↓ 15 callersFunctioncast
(conn=None, cmd=None, msg=None)
python/oneflow/framework/unittest.py:167
↓ 15 callersMethodlower
oneflow/user/kernels/gather_kernel.cpp:36
↓ 15 callersMethodnccl_use_compute_stream
oneflow/core/job/resource_desc.cpp:73
↓ 15 callersMethodreset
oneflow/extension/stack/stacktrace.h:601
↓ 15 callersMethodshape_view
oneflow/ir/include/OneFlow/OKL/Kernel/TmpBufferManager.h:36
↓ 15 callersMethodupper
oneflow/user/kernels/gather_kernel.cpp:37
↓ 14 callersFunctionAsStrided
oneflow/core/framework/tensor_methods.cpp:681
↓ 14 callersFunctionCheckDataTypeLike
oneflow/user/ops/model_update_ops.cpp:31
↓ 14 callersFunctionConnect
oneflow/core/graph/node.h:25
↓ 14 callersFunctionError
oneflow/maybe/maybe.h:186
↓ 14 callersMethodError
oneflow/maybe/maybe.h:69
↓ 14 callersFunctionGetDefaultDType
oneflow/core/framework/dtype.cpp:272
↓ 14 callersFunctionGetNdSbpStrList
oneflow/core/framework/nd_sbp.cpp:68
↓ 14 callersFunctionGetOpTypeName
oneflow/ir/include/OneFlow/OneFlowOps.h:52
↓ 14 callersMethodGetYShape5D
oneflow/user/kernels/max_pool_kernel_util.cpp:108
↓ 14 callersMethodHasRegstDescId
oneflow/core/lazy/actor/register_slot.cpp:27
↓ 14 callersFunctionHash
oneflow/core/common/hash.h:35
↓ 14 callersFunctionIsHostMem
oneflow/core/memory/memory_case_util.cpp:108
↓ 14 callersMethodIsTrain
oneflow/core/job/job_desc.h:51
↓ 14 callersFunctionIsTriviallyCopyableDataType
oneflow/core/common/data_type.cpp:68
↓ 14 callersFunctionIsValidChainId
oneflow/core/graph/task_node.h:44
← previousnext →501–600 of 28,456, ranked by callers