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
↓ 14 callersMethodNdIndexToOffset
oneflow/core/ep/common/primitive/broadcast_elementwise_unary.h:63
↓ 14 callersMethodWaitForeverUntilCntEqualZero
oneflow/core/common/blocking_counter.cpp:48
↓ 14 callersMethod__init__
(self, reduction: str = "mean")
python/oneflow/nn/modules/loss.py:25
↓ 14 callersMethod__setattr__
(self, name: str, value: Union[Tensor, "Module"])
python/oneflow/nn/modules/module.py:437
↓ 14 callersFunction_assert_true
(test_case, value1, value2)
python/oneflow/test/modules/test_global_higher_derivative_activation.py:28
↓ 14 callersFunction_assert_true
(test_case, value1, value2)
python/oneflow/test/modules/test_higher_derivative_activation.py:28
↓ 14 callersFunction_test_fused_scale_mask_bias_softmax
( test_case, N=512, S=128, D=128, h=8, d=32, mode="row", ensemble_batch=8,
python/oneflow/test/modules/test_fused_scale_mask_bias_softmax.py:66
↓ 14 callersFunction_test_layer_norm
( test_case, shape, normalized_shape, affine=True, eps=1e-6, dtype=flow.float32, d
python/oneflow/test/modules/test_layer_norm.py:48
↓ 14 callersMethodcompress_info
oneflow/user/image/jpeg_decoder.cpp:30
↓ 14 callersMethodcumsum
(sequence)
python/oneflow/utils/data/dataset.py:224
↓ 14 callersMethoddata_type
oneflow/ir/include/OneFlow/OKL/Kernel/TmpBufferManager.h:38
↓ 14 callersMethoddrop
oneflow/maybe/just_test.cpp:229
↓ 14 callersMethoddtype
oneflow/api/cpp/framework/tensor.cpp:67
↓ 14 callersFunctionfloat_from_bytes
oneflow/core/common/bfloat16_test.cpp:23
↓ 14 callersMethodis_lazy
oneflow/core/framework/tensor.h:370
↓ 14 callersMethodmemory_format
oneflow/core/common/tensor_meta.h:51
↓ 14 callersMethodmut_status_buffer
oneflow/core/vm/instruction.h:134
↓ 14 callersMethodnamed_parameters
r""" named_parameters(prefix="", recurse=True) -> Iterator[Tuple[str, Tensor]] Returns an iterator over module parameters, yi
python/oneflow/nn/modules/module.py:551
↓ 14 callersMethodoutput_arg_tuple
oneflow/user/kernels/stateful_opkernel.h:89
↓ 14 callersMethodprint
oneflow/ir/lib/OneFlow/OneFlowOps.cpp:295
↓ 14 callersMethodptr
oneflow/core/embedding/persistent_table.cpp:138
↓ 14 callersMethodregister_num
oneflow/core/register/runtime_register_desc.h:36
↓ 14 callersMethodset_foo_map_key
oneflow/core/intrusive/skiplist_test.cpp:43
↓ 14 callersFunctiontrunc_normal_
(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0)
python/oneflow/nn/init.py:256
↓ 13 callersFunctionCreateTransposeValue
oneflow/ir/lib/OneFlow/Conversion/OneFlowToTosa.cpp:105
↓ 13 callersFunctionInferNdSbp4SrcOp
oneflow/core/framework/sbp_context.cpp:118
↓ 13 callersMethodIntersect
oneflow/core/register/tensor_slice_view.cpp:75
↓ 13 callersMethodIsAsyncLaunchCclLogicalKernel
oneflow/core/job/eager_nccl_comm_manager.h:41
↓ 13 callersFunctionNdSbpToStringList
oneflow/core/job/sbp_parallel.cpp:259
↓ 13 callersMethodNext
oneflow/core/intrusive/list.h:59
↓ 13 callersMethodParallelId4MachineDeviceId
oneflow/core/job/parallel_desc.cpp:146
↓ 13 callersMethodParse
oneflow/user/data/coco_parser.cpp:23
↓ 13 callersFunctionParseSbpParallelFromString
oneflow/core/job/sbp_parallel.cpp:217
↓ 13 callersMethodRecord
oneflow/core/framework/tensor_name_scope.cpp:46
↓ 13 callersFunctionUseDynamicMemoryAllocation
oneflow/core/embedding/embedding_manager.h:29
↓ 13 callersFunctionValue
oneflow/maybe/optional.h:267
↓ 13 callersMethod__init__
(self)
python/oneflow/test/modules/test_module.py:362
↓ 13 callersMethod__init__
( self, kernel_size: _size_1_t, stride: Optional[_size_1_t] = None, padding: _
python/oneflow/nn/modules/pooling.py:84
↓ 13 callersMethodadd_pruning_method
r"""Adds a child pruning ``method`` to the container. Args: method (subclass of BasePruningMethod): child pruning method
python/oneflow/nn/utils/prune.py:291
↓ 13 callersMethodbar
oneflow/core/intrusive/intrusive_core_test.cpp:56
↓ 13 callersMethodbroadcast_like
(x, y, *args, **kwargs)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:1063
↓ 13 callersFunctioncall
(conn=None, cmd=None, msg=None)
python/oneflow/framework/unittest.py:174
↓ 13 callersMethoddptr
oneflow/core/framework/user_op_tensor.h:52
↓ 13 callersMethoddst_actor_id
oneflow/core/lazy/actor/actor_message.h:46
↓ 13 callersFunctiongetNHWCResultTypes
oneflow/ir/lib/OneFlow/TransposeHelpers.cpp:37
↓ 13 callersFunctiongetOpName
oneflow/ir/lib/OneFlow/OneFlowOpTraits.cpp:95
↓ 13 callersFunctionis_global
()
python/oneflow/test_utils/automated_test_util/global_scope.py:34
↓ 13 callersMethodkey
(self)
python/oneflow/profiler/events.py:108
↓ 13 callersMethodmut_shape_view
oneflow/core/register/blob.h:104
↓ 13 callersMethodrelease
oneflow/api/python/multiprocessing/object_ptr.h:37
↓ 13 callersFunctionset_enabled
oneflow/core/framework/autocast.cpp:75
↓ 13 callersFunctionsin
oneflow/core/common/bfloat16_math.h:42
↓ 13 callersMethodstart
oneflow/core/functional/tensor_index.h:47
↓ 13 callersMethodstate
oneflow/core/framework/op_expr.h:337
↓ 13 callersMethodswap
oneflow/extension/stack/stacktrace.h:620
↓ 13 callersMethodthrd_id
oneflow/core/graph/task_node.h:56
↓ 12 callersFunctionCheckLearningRateDataType
oneflow/user/ops/model_update_ops.cpp:59
↓ 12 callersFunctionCheckLearningRateShape
oneflow/user/ops/model_update_ops.cpp:52
↓ 12 callersMethodClose
oneflow/core/common/buffer.h:79
↓ 12 callersMethodElemNum
oneflow/core/ndarray/xpu_shape.h:42
↓ 12 callersMethodEnd
oneflow/core/intrusive/list.h:69
↓ 12 callersMethodEqualsEnd
oneflow/core/intrusive/skiplist.h:59
↓ 12 callersFunctionGenOpBlobArg
oneflow/core/operator/operator.h:420
↓ 12 callersFunctionGetNearestInputIndex
oneflow/user/kernels/upsample_kernel.h:29
↓ 12 callersMethodGetOpParallelDesc
oneflow/core/operator/operator.cpp:157
↓ 12 callersFunctionGetStageIdHint
oneflow/core/job_rewriter/pipeline_buffer_pass.cpp:83
↓ 12 callersFunctionInitPoolOpKernelCache
oneflow/user/kernels/tf_pool_cpu_kernel.cpp:31
↓ 12 callersMethodMatch
oneflow/core/intrusive/flat_msg_view.h:333
↓ 12 callersFunctionRingDecrease
oneflow/user/kernels/collective_communication/cpu/cpu_collective_communication_util.h:26
↓ 12 callersFunctionSbpParallelToString
oneflow/core/job/sbp_parallel.cpp:248
↓ 12 callersMethodShapeView4ArgNameAndIndex
oneflow/core/kernel/user_kernel.cpp:491
↓ 12 callersMethodStackSize
oneflow/maybe/error.h:255
↓ 12 callersMethodTensor4ArgNameAndIndex
oneflow/core/kernel/user_kernel.cpp:486
↓ 12 callersMethod__init__
(self)
python/oneflow/test/graph/test_graph.py:32
↓ 12 callersFunction_check_equal
(test_case, lhs, rhs, name="", rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_higher_derivative_pool.py:24
↓ 12 callersFunction_check_equal
(test_case, lhs, rhs, name="", rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_global_higher_derivative_pool.py:24
↓ 12 callersFunction_test_broadcast_rand_op_in_graph
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/graph/test_graph_random_seed.py:220
↓ 12 callersFunction_test_fold_multiply
(test_case, module, with_cuda, *args, dtype=oneflow.float32)
oneflow/ir/test/OneFlow/folding/test_simple_multiply.py:56
↓ 12 callersFunction_test_fused_linear
(test_case, num_bits, m, k, n, group_dim, group_size)
python/oneflow/test/modules/test_groupwise_quantization.py:138
↓ 12 callersFunction_test_global_rand_op_with_broadcast
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/modules/test_random_generator_and_seed.py:106
↓ 12 callersFunction_test_global_rand_op_with_split
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/modules/test_random_generator_and_seed.py:78
↓ 12 callersFunction_test_rand_op_in_FRB
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/graph/test_graph_random_seed.py:120
↓ 12 callersFunction_test_rand_op_in_graph
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/graph/test_graph_random_seed.py:50
↓ 12 callersFunction_test_rand_op_unidentical
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/modules/test_random_generator_and_seed.py:54
↓ 12 callersFunction_test_split_rand_op_in_graph
(test_case, rand_op, input=None, **kwargs)
python/oneflow/test/graph/test_graph_random_seed.py:173
↓ 12 callersFunction_to_layout
(ts, layout, tensor_index, seq_len=None)
python/oneflow/test/modules/test_fused_attention_ops.py:82
↓ 12 callersFunctionconvertAscendType
oneflow/ir/oneflow-lite/include/OneFlow/Transform/Lowering/LoweringAscendUtils.h:100
↓ 12 callersMethodcudnn_handle
oneflow/core/device/cudnn_conv_util.cpp:354
↓ 12 callersMethodempty
oneflow/api/python/caster/size.h:34
↓ 12 callersMethoderror
oneflow/core/common/maybe.h:192
↓ 12 callersMethodfd
oneflow/core/embedding/posix_file.h:62
↓ 12 callersFunctionfft_norm_from_string
Convert NumPy compatible normalization mode string to enum values In Numpy, "forward" translates to `by_n` for a forward transform and `none` for back
oneflow/core/functional/impl/math_functor.cpp:3925
↓ 12 callersFunctionget_fake_program_more_detail
(oneflow, mode, func, args=None, kwargs=None)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:280
↓ 12 callersFunctionhrcp
oneflow/core/device/cuda_pseudo_half.h:147
↓ 12 callersMethodlowerOp
oneflow/ir/oneflow-lite/lib/OneFlow/Transform/Lowering/LoweringAscend.cpp:170
↓ 12 callersMethodmachine_id
Getters
oneflow/core/graph/task_node.h:55
↓ 12 callersMethodrematable
oneflow/core/framework/device.h:46
↓ 12 callersMethodto
( self: T, device: Optional[Union[int, str, flow.device]] = ..., dtype: Optional[flow.
python/oneflow/nn/modules/module.py:1446
↓ 11 callersFunctionBasicView
oneflow/core/framework/tensor_methods.cpp:82
↓ 11 callersMethodClear
oneflow/core/intrusive/list.h:130
← previousnext →601–700 of 28,456, ranked by callers