MCPcopy Create free account

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

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

↓ 5 callersMethodSwap
oneflow/core/common/tensor_buffer.cpp:113
↓ 5 callersFunctionSymmetricNDimToNDimBoxingExpr
oneflow/core/boxing/eager_boxing_interpreter_mgr.cpp:67
↓ 5 callersFunctionTensorDescInferFn
oneflow/user/ops/scalar_by_tensor_op.cpp:23
↓ 5 callersFunctionTensorDeviceToString
oneflow/core/functional/impl/binary_functor.cpp:50
↓ 5 callersFunctionTensorSize_Check
oneflow/api/python/framework/size.cpp:168
↓ 5 callersMethodTryReceive
oneflow/core/common/buffer.h:69
↓ 5 callersFunctionUnpackIndexItem
oneflow/api/python/functional/indexing.cpp:196
↓ 5 callersFunctionUpdatePlacement
oneflow/ir/oneflow-translate/lib/OneFlow/MLIROneFlowTranslation.cpp:603
↓ 5 callersMethodValue
oneflow/maybe/maybe.h:65
↓ 5 callersFunctionWithOpGraphAndMutJobBuilder
oneflow/core/job_rewriter/job_completer.cpp:72
↓ 5 callersMethodWithScopedRelease
oneflow/core/common/foreign_lock_helper.cpp:21
↓ 5 callersFunction_
(mod: torch.nn.Module, verbose=False)
python/oneflow/framework/infer_compiler/transform/builtin_transform.py:190
↓ 5 callersMethod__init__
(self, torch_module, oneflow_module)
python/oneflow/framework/infer_compiler/with_oneflow_compile.py:36
↓ 5 callersMethod__init__
(self, data)
python/oneflow/test/graph/test_graph_pipeline.py:201
↓ 5 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: _size_1_t, s
python/oneflow/nn/modules/conv.py:188
↓ 5 callersFunction_add_conv
( out, in_channels, channels, kernel=1, stride=1, pad=0, num_group=1, active=T
python/oneflow/test/expensive/pytorch_rexnetv1_lite.py:47
↓ 5 callersFunction_check_forward_and_backward
(test_case, input, of_out, torch_out)
python/oneflow/test/modules/test_global_slice.py:25
↓ 5 callersFunction_fill_in_zeros
(grads, refs, strict, create_graph, stage)
python/oneflow/autograd/functional.py:207
↓ 5 callersMethod_generate_config_proto
(self)
python/oneflow/nn/graph/graph.py:743
↓ 5 callersFunction_generate_output_size
(input_size, output_size)
python/oneflow/nn/modules/utils.py:101
↓ 5 callersMethod_get_abs_string_index
Get the absolute index for the list of modules
python/oneflow/nn/utils/container.py:112
↓ 5 callersFunction_inspect_rand_op_and_args
(rand_op, **kwargs)
python/oneflow/test/graph/test_graph_random_seed.py:26
↓ 5 callersFunction_np_get_expand
(input_shape, expand_size)
python/oneflow/test/modules/test_expand.py:27
↓ 5 callersFunction_test_bias_add_fused_scale_mask_softmax_dropout
( test_case, input_shape, bias_shape, mask_shape, input_dtype=flow.float32, mask_dtype
python/oneflow/test/modules/test_fused_bias_add_scale_mask_softmax_dropout.py:37
↓ 5 callersFunction_test_tensor_load_h2d
(test_case, input, tensor_mem)
python/oneflow/test/expensive/test_tensor_offload.py:59
↓ 5 callersFunction_test_tensor_offload_d2h
(test_case, input, tensor_mem)
python/oneflow/test/expensive/test_tensor_offload.py:32
↓ 5 callersFunction_try_convert_to_local_tensor
(tensor)
python/oneflow/framework/tensor_str_util.py:32
↓ 5 callersMethodadd_stack_frame
Setters
oneflow/core/common/error.h:120
↓ 5 callersMethodbind
oneflow/core/common/optional.h:420
↓ 5 callersFunctioncached_casts
oneflow/core/framework/autocast.cpp:67
↓ 5 callersFunctioncheck_rnn_cell_forward_hidden
oneflow/core/functional/impl/rnn_functor.cpp:68
↓ 5 callersMethodchildren
r""" children() -> Iterator["Module"] Returns an iterator over immediate children modules. Yields: Modul
python/oneflow/nn/modules/module.py:637
↓ 5 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_fused_get_ciou_diagonal_angle.py:36
↓ 5 callersFunctioncompare
(a, b, rtol=1e-5, atol=1e-5)
python/oneflow/test/modules/test_fused_get_ciou_result.py:60
↓ 5 callersMethodcompute_op
oneflow/core/eager/tensor_storage.cpp:117
↓ 5 callersFunctionconvertStrides
oneflow/ir/oneflow-lite/include/OneFlow/Transform/Lowering/LoweringAscendUtils.h:144
↓ 5 callersMethodcopy_to
oneflow/api/cpp/framework/tensor.cpp:104
↓ 5 callersMethodcreateInt64Vec
oneflow/ir/oneflow-lite/include/OneFlow/FlatbufferUtils.h:114
↓ 5 callersMethoddevice_type_name
oneflow/core/framework/auto_random_generator.h:38
↓ 5 callersFunctiondie_has_pc
oneflow/extension/stack/stacktrace.h:2689
↓ 5 callersMethodelem_cnt
oneflow/api/cpp/framework/shape.cpp:49
↓ 5 callersMethodempty
oneflow/core/intrusive/list_hook.h:39
↓ 5 callersMethodenable_auto_parallel_ignore_user_sbp_config
If true, it will ignore all user configurations of SBP. Args: mode (bool, optional): [description]. Default is True.
python/oneflow/nn/graph/graph_config.py:384
↓ 5 callersMethodep_event_provider
oneflow/core/vm/ep_stream_policy_base.h:49
↓ 5 callersMethodexec_module
(self, module)
python/oneflow/mock_torch/mock_importer.py:166
↓ 5 callersMethodforced_eviction_num
oneflow/core/vm/remat/env.h:52
↓ 5 callersFunctionformat_time
(time_us)
python/oneflow/profiler/util.py:20
↓ 5 callersMethodget
oneflow/core/intrusive/shared_ptr.h:52
↓ 5 callersFunctiongetAttrAsShape
oneflow/ir/lib/OneFlow/UserOpConversion.cpp:422
↓ 5 callersFunctionget_cost
oneflow/core/vm/remat/allocator.cpp:140
↓ 5 callersMethodget_id
oneflow/ir/oneflow-extension/include/PyAst/Ast.h:407
↓ 5 callersMethodget_lr
(self, base_lr, step)
python/oneflow/nn/optimizer/step_lr.py:63
↓ 5 callersFunctionget_proxy_cls
(item)
python/oneflow/nn/graph/proxy.py:41
↓ 5 callersFunctionget_referenced_die_name
oneflow/extension/stack/stacktrace.h:2649
↓ 5 callersFunctionglob_by_pattern
(pattern)
tools/generate_header_list.py:11
↓ 5 callersMethodglobal_tensor_meta
oneflow/core/framework/tensor.h:74
↓ 5 callersMethodgrad_fn_node
Getters for autograd acc_grad is tensor's accumulated grad in more than once backward operation, and current_grad is temporary grad to shared data wit
oneflow/core/framework/tensor.h:334
↓ 5 callersMethodis_alive
(self)
python/oneflow/utils/data/_utils/worker.py:65
↓ 5 callersMethodis_evictable
oneflow/core/eager/tensor_storage.cpp:110
↓ 5 callersMethodis_view
oneflow/core/framework/tensor.h:428
↓ 5 callersFunctionmake_constant_initializer
make constant initializer param of make_table_options Args: constant (float): A python scalar. value to generate. Returns: d
python/oneflow/one_embedding.py:771
↓ 5 callersMethodmake_contiguous_params_group
r"""Get contiguous parameters group after creating the whole module. Rearrange the parameters of the model in the same dtype and device
python/oneflow/nn/modules/module.py:1804
↓ 5 callersMethodmul
oneflow/core/ep/common/primitive/broadcast_elementwise_unary.h:161
↓ 5 callersMethodmut_device
oneflow/core/vm/ep_event.h:35
↓ 5 callersMethodmut_event
oneflow/core/vm/ep_event.h:37
↓ 5 callersMethodmut_output_blob_time_shape
oneflow/core/operator/user_op.cpp:501
↓ 5 callersMethodmut_raw_dptr
oneflow/core/register/blob.h:94
↓ 5 callersFunctionnaive_embedding
( x, cos, sin, x_layout, B, M, H, K, dims, merged_dims, rotary_siz
python/oneflow/test/modules/test_fused_rotary_embedding.py:92
↓ 5 callersMethodname
r"""Name auto-generated for this graph.
python/oneflow/nn/graph/graph.py:466
↓ 5 callersMethodnccl_comm
oneflow/user/kernels/collective_communication/cuda/cuda_communication_context.h:38
↓ 5 callersMethodop_node
op_node_
oneflow/core/graph/compute_task_node.h:48
↓ 5 callersMethodparallel_desc
oneflow/user/kernels/stateful_opkernel.cpp:84
↓ 5 callersMethodparallel_desc
oneflow/user/kernels/collective_communication/cpu/cpu_communication_context.h:35
↓ 5 callersFunctionparseDims
(dims, x_layout)
python/oneflow/test/modules/test_fused_rotary_embedding.py:39
↓ 5 callersFunctionprint_note_fake_program
(detail=False)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:904
↓ 5 callersFunctionproduct
(tu)
python/oneflow/test/modules/test_quantization.py:56
↓ 5 callersMethodptr
oneflow/core/intrusive/force_standard_layout_test.cpp:43
↓ 5 callersMethodregister_forward_hook
r""" register_forward_hook(hook) Registers a forward hook on the module. The hook will be called every time after :func:`for
python/oneflow/nn/modules/module.py:1248
↓ 5 callersMethodregister_forward_pre_hook
r""" register_forward_pre_hook(hook) Registers a forward pre-hook on the module. The hook will be called every time
python/oneflow/nn/modules/module.py:1226
↓ 5 callersFunctionremove
r"""Removes the pruning reparameterization from a module and the pruning method from the forward hook. The pruned parameter named ``name`` rem
python/oneflow/nn/utils/prune.py:1130
↓ 5 callersMethodrun_cnt
oneflow/core/framework/nn_graph.h:72
↓ 5 callersFunctionrun_testcase_with_sep_compile
(test_case_cls)
python/oneflow/test/graph/test_graph_separate_compile.py:58
↓ 5 callersMethodset_mem_block_offset
oneflow/core/register/register_desc.h:70
↓ 5 callersMethodshared_op
oneflow/core/graph/op_graph.h:43
↓ 5 callersFunctionspawn_shell_ignoring_failure
(cmd: str = None)
ci/test/distributed_run.py:69
↓ 5 callersMethodsrc_placed_nd_sbp
oneflow/core/boxing/boxing_interpreter_status.h:65
↓ 5 callersMethodstream_sequential_dependence
oneflow/core/vm/instruction_policy.h:45
↓ 5 callersMethodthread_unsafe_size
oneflow/core/intrusive/mutexed_list.h:37
↓ 5 callersMethodtransport_token
oneflow/core/framework/tensor.h:414
↓ 5 callersMethodupdate
(self, *args, **kwargs)
python/oneflow/test/graph/test_graph_io_check.py:71
↓ 5 callersMethodupdate_lrs
(self, lrs)
python/oneflow/nn/optimizer/lr_scheduler.py:71
↓ 5 callersMethodvalue_type
oneflow/api/python/functional/python_arg.h:44
↓ 5 callersMethodworld_size
oneflow/core/control/ctrl_bootstrap.h:60
↓ 4 callersMethodAbort
oneflow/maybe/error.h:222
↓ 4 callersMethodAddBnToRegstAndBindIt
oneflow/core/graph/exec_graph.cpp:32
↓ 4 callersFunctionAddFreeEagerTensorToVariableOp
oneflow/core/framework/op_interpreter/lazy_op_interpreter.cpp:205
↓ 4 callersFunctionAddLbns
oneflow/core/job_rewriter/autograd.cpp:279
↓ 4 callersMethodAllocate
oneflow/core/vm/bin_allocator.h:381
↓ 4 callersMethodAllocate
oneflow/core/vm/bin_allocator_test.cpp:40
↓ 4 callersMethodAppendQueue
oneflow/user/summary/events_writer.cpp:64
← previousnext →1,301–1,400 of 28,456, ranked by callers