Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Oneflow-Inc/oneflow
/ functions
Functions
28,456 in github.com/Oneflow-Inc/oneflow
⨍
Functions
28,456
◇
Types & classes
6,969
↳
Endpoints
179
↓ 34 callers
Method
AddLbi
oneflow/core/graph/task_node.h:194
↓ 34 callers
Function
DeviceTag4DeviceType
oneflow/core/framework/to_string.cpp:23
↓ 34 callers
Method
Front
oneflow/core/lazy/actor/register_slot.cpp:70
↓ 34 callers
Method
NumAxes
oneflow/core/ndarray/xpu_shape.h:43
↓ 34 callers
Method
apply
r"""Adds the forward pre-hook that enables pruning on the fly and the reparametrization of a tensor in terms of the original tensor an
python/oneflow/nn/utils/prune.py:427
↓ 34 callers
Method
debug
r"""Open or close debug mode of the graph. If in debug mode, logs of computation graph building infos or warnings will be printed. Ot
python/oneflow/nn/graph/graph.py:481
↓ 34 callers
Function
ri
(indices)
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:229
↓ 34 callers
Method
values
Return an iterable of the ModuleDict values.
python/oneflow/nn/utils/container.py:246
↓ 33 callers
Method
Build
Explicitly complete the construction of the builtin op Returns: the completed builtin op
python/oneflow/ops/stateful_ops.py:68
↓ 33 callers
Function
IsIntegralDataType
oneflow/core/common/data_type.cpp:31
↓ 33 callers
Function
PbMessage2TxtString
oneflow/core/common/protobuf.cpp:54
↓ 33 callers
Function
g
(x)
python/oneflow/test/exceptions/throw_error.py:23
↓ 33 callers
Method
storage_offset
oneflow/core/framework/tensor.h:83
↓ 32 callers
Function
AutoMemcpy
oneflow/core/kernel/kernel_util.cpp:27
↓ 32 callers
Method
GetXShape5D
oneflow/user/kernels/max_pool_kernel_util.cpp:104
↓ 32 callers
Method
Output
Set output blob of op Args: output_name (str): name of output blob num (int, optional): Defaults to 1. Retu
python/oneflow/ops/stateful_ops.py:54
↓ 32 callers
Method
__print
r"""Do print according to info level.
python/oneflow/nn/graph/graph.py:630
↓ 32 callers
Method
is_eager
oneflow/core/framework/tensor.h:66
↓ 32 callers
Function
is_in_memory
(tensor)
python/oneflow/test/expensive/_test_remat.py:37
↓ 31 callers
Function
LocalToGlobal
oneflow/core/functional/impl/global_cast.cpp:415
↓ 31 callers
Method
_shallow_repr
(self)
python/oneflow/nn/graph/graph.py:611
↓ 31 callers
Method
end
oneflow/api/python/caster/size.h:38
↓ 31 callers
Function
log
oneflow/core/common/bfloat16_math.h:32
↓ 31 callers
Method
mutable_sbp_signature
oneflow/core/operator/user_op.cpp:450
↓ 31 callers
Method
next
oneflow/core/intrusive/list_hook.h:31
↓ 31 callers
Method
register_buffer
r""" register_buffer(name, tensor, persistent=True) Adds a buffer to the module. This is typically used to register
python/oneflow/nn/modules/module.py:277
↓ 31 callers
Function
torch2oflow
(mod, *args, **kwargs)
python/oneflow/framework/infer_compiler/transform/builtin_transform.py:162
↓ 30 callers
Method
Begin
oneflow/core/intrusive/list.h:49
↓ 30 callers
Function
IsUserOpWithTypeName
oneflow/core/job_rewriter/pass_util.cpp:74
↓ 30 callers
Function
Narrow
oneflow/core/framework/tensor_methods.cpp:515
↓ 30 callers
Method
NewNode
oneflow/core/graph/graph.h:314
↓ 30 callers
Function
PyTensor_New
oneflow/api/python/framework/tensor.cpp:854
↓ 30 callers
Method
ToDimVector
oneflow/core/common/shape_view.cpp:22
↓ 30 callers
Method
__init__
(self)
python/oneflow/test/graph/test_graph_block.py:284
↓ 30 callers
Function
_cpu_global_tensor
(tensor)
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:54
↓ 30 callers
Method
data
oneflow/core/common/tensor_buffer.h:110
↓ 30 callers
Method
is_floating_point
oneflow/core/framework/dtype.cpp:152
↓ 30 callers
Method
min
oneflow/core/common/bfloat16.h:283
↓ 30 callers
Method
output_size
oneflow/core/framework/user_op_conf.cpp:78
↓ 29 callers
Function
DefaultAutoGenerator
oneflow/core/framework/random_generator.cpp:136
↓ 29 callers
Method
Erase
oneflow/core/intrusive/list.h:110
↓ 29 callers
Method
IsIntegral
oneflow/core/common/scalar.h:89
↓ 29 callers
Method
IsOk
oneflow/core/common/maybe.h:187
↓ 29 callers
Method
eval
(self)
python/oneflow/test_utils/automated_test_util/generators.py:95
↓ 29 callers
Method
getOpName
oneflow/ir/include/OneFlow/OneFlowOpTraits.h:60
↓ 29 callers
Function
global_broadcast_consec
Generate a arithmetic progression with given size and start value.
python/oneflow/test/tensor/test_lazy_tensor_indexing.py:65
↓ 29 callers
Method
mut_head
Setters
oneflow/core/intrusive/list_test.cpp:307
↓ 29 callers
Method
remove
(self)
python/oneflow/utils/hooks.py:39
↓ 29 callers
Method
shape
oneflow/api/python/caster/size.h:30
↓ 29 callers
Method
tensor_storage
oneflow/core/framework/tensor.h:81
↓ 28 callers
Function
DeviceAndStreamInferFn
oneflow/user/ops/comm_net_device_infer_util.h:34
↓ 28 callers
Function
Expand
oneflow/core/framework/tensor_methods.cpp:394
↓ 28 callers
Function
SetInputArgModifierMutable
oneflow/user/ops/model_update_ops.cpp:339
↓ 28 callers
Function
Unsqueeze
oneflow/core/framework/tensor_methods.cpp:220
↓ 28 callers
Function
compare_result
(a, b, rtol=1e-5, atol=1e-8)
python/oneflow/test/tensor/test_complex.py:69
↓ 28 callers
Method
ref_cnt
oneflow/core/intrusive/ref.h:33
↓ 28 callers
Method
run
tools/oneflow-tblgen/op_schema_emitter.cpp:110
↓ 27 callers
Function
GetInDim
oneflow/core/operator/operator_util.cpp:45
↓ 27 callers
Function
HashCombine
oneflow/maybe/utility.h:69
↓ 27 callers
Function
SwitchCase
oneflow/core/common/switch_func.h:24
↓ 27 callers
Function
_global_math_op_grad_grad_impl
(test_case, op_name, placement, sbp)
python/oneflow/test/modules/test_global_math_op_higher_derivative.py:24
↓ 27 callers
Function
_is_raw_type
(value, raw_type)
python/oneflow/framework/args_tree.py:21
↓ 27 callers
Function
_test_math_op_grad_grad_impl
(test_case, op_name)
python/oneflow/test/modules/test_math_op_higher_derivative.py:24
↓ 27 callers
Method
add_module
r""" add_module(name, module) Adds a child module to the current module. The module can be accessed as an attribute
python/oneflow/nn/modules/module.py:252
↓ 27 callers
Method
device
oneflow/api/cpp/framework/tensor.cpp:62
↓ 27 callers
Method
is_complex
oneflow/core/framework/dtype.cpp:117
↓ 27 callers
Method
map
oneflow/core/common/optional.h:410
↓ 27 callers
Method
reset
oneflow/core/framework/mutable_attr_map.h:53
↓ 26 callers
Function
CastToPyObject
oneflow/api/python/functional/common.h:205
↓ 26 callers
Method
GetSoleRegst
oneflow/core/graph/task_node.cpp:418
↓ 26 callers
Function
IsFloatingDataType
oneflow/core/common/data_type.cpp:40
↓ 26 callers
Method
Lookup
oneflow/core/lazy/actor/light_actor.cpp:129
↓ 26 callers
Function
MakeHostMemCase
oneflow/core/memory/memory_case_util.cpp:101
↓ 26 callers
Method
Set
oneflow/core/ndarray/xpu_shape.h:48
↓ 26 callers
Function
Transpose
oneflow/core/framework/tensor_methods.cpp:735
↓ 26 callers
Method
Write
oneflow/core/persistence/tee_persistent_log_stream.cpp:42
↓ 26 callers
Method
__init__
(self, alpha: float = 1.0, inplace: bool = False)
python/oneflow/nn/modules/activation.py:254
↓ 26 callers
Function
compare_result
(test_case, a, b, rtol=1e-5, atol=1e-8)
python/oneflow/test/modules/test_fused_glu.py:123
↓ 26 callers
Method
length
oneflow/user/kernels/pocketfft_hdronly.h:2510
↓ 26 callers
Method
parse_args
oneflow/extension/stack/stacktrace.h:689
↓ 26 callers
Function
random_dtype
(seq_names)
python/oneflow/test_utils/automated_test_util/torch_flow_dual_object.py:1437
↓ 26 callers
Function
spawn_shell
(cmd: str = None)
ci/test/distributed_run.py:63
↓ 25 callers
Method
ByteSizeOfBlobBody
oneflow/core/register/blob.h:121
↓ 25 callers
Method
Decrease
oneflow/core/common/blocking_counter.cpp:31
↓ 25 callers
Function
GetModule
oneflow/ir/lib/OneFlow/OKL/Kernel/LauncherState.cpp:41
↓ 25 callers
Function
GetValidMaxCopyCost
oneflow/core/framework/sbp_infer_util.cpp:746
↓ 25 callers
Method
IsFloatingPoint
oneflow/core/common/scalar.h:90
↓ 25 callers
Function
RegisterBoxingFunction
oneflow/core/boxing/eager_boxing_interpreter.h:59
↓ 25 callers
Function
TestEnumerateNdSbpIn2OutSignatures
oneflow/user/ops/reshape_user_op_util_test.cpp:68
↓ 25 callers
Method
TryPushBackRegst
oneflow/core/lazy/actor/register_slot.cpp:37
↓ 25 callers
Method
__contains__
(self, key: str)
python/oneflow/nn/utils/container.py:218
↓ 25 callers
Method
begin
oneflow/api/python/caster/size.h:37
↓ 25 callers
Method
data_type
oneflow/core/kernel/blob_tensor_view.cpp:31
↓ 25 callers
Method
enable_reuse_mem
oneflow/core/job/job_desc.h:53
↓ 25 callers
Method
parallel_num
oneflow/core/framework/op_expr.cpp:494
↓ 25 callers
Method
rank
oneflow/core/graph/stream_id.h:42
↓ 25 callers
Method
retain_grad
oneflow/core/framework/tensor.h:426
↓ 25 callers
Method
to
Performs dtype and/or device conversion on `self.data`. It has similar signature as :meth:`oneflow.Tensor.to`, except optional argume
python/oneflow/nn/utils/rnn.py:162
↓ 25 callers
Method
update
(self, event)
python/oneflow/profiler/events.py:47
↓ 24 callers
Function
GetPhysicalShape
oneflow/core/operator/operator.cpp:1654
← previous
next →
301–400 of 28,456, ranked by callers