Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/MegEngine/MegEngine
/ functions
Functions
33,912 in github.com/MegEngine/MegEngine
⨍
Functions
33,912
◇
Types & classes
7,186
↳
Endpoints
66
↓ 26 callers
Method
run
r""" Args: inp_args: list of input datas. inp_dict: dict of named input datas. Returns: a dict {
imperative/python/megengine/utils/comp_graph_tools.py:465
↓ 26 callers
Method
set
dnn/src/cuda/dropout/opr_impl.h:13
↓ 26 callers
Method
set
dnn/src/cuda/convolution/helper.h:33
↓ 26 callers
Function
sizeof_fmt
(num, suffix="B")
imperative/python/megengine/utils/module_stats.py:198
↓ 26 callers
Function
tester
(OptCls, **kwargs)
imperative/python/test/unit/xla/test_xla_training.py:156
↓ 25 callers
Function
_compare_nchw_nhwc
(data, func, is_symbolic=None)
imperative/python/test/unit/core/test_formatted_tensor.py:44
↓ 25 callers
Method
backward
(self, dy)
imperative/python/test/unit/autodiff/test_grad_manager.py:436
↓ 25 callers
Method
code_gen
src/jit/include/megbrain/jit/ast_c.h:55
↓ 25 callers
Method
define_bool_state
Set up thread-local state and return a contextmanager for managing it. This function is a convenience wrapper. It defines a flag, environment
imperative/python/megengine/xla/lib/config.py:141
↓ 25 callers
Method
dispatch_on_comp_node
src/core/impl/graph/normal_exec_env.cpp:104
↓ 25 callers
Method
get_used_memory
src/core/impl/comp_node/cuda/comp_node.cpp:537
↓ 25 callers
Method
get_var
src/jit/impl/executor_opr.cpp:440
↓ 25 callers
Method
idx
get logical index; valonly must be false
dnn/include/megdnn/tensor_iter.h:135
↓ 25 callers
Function
incr_refp
dnn/src/common/utils.h:361
↓ 25 callers
Function
interleave_helper
dnn/src/aarch64/matrix_mul/asm/common.h:1272
↓ 25 callers
Function
is_static_var_value
! * \brief whether var value could be statically inferred */
src/core/include/megbrain/graph/helper.h:40
↓ 25 callers
Method
item
r"""Returns the value of this :class:`~.Tensor` as a standard Python :class:`numbers.Number`. This only works for tensors with one element. Fo
imperative/python/megengine/core/tensor/array_method.py:371
↓ 25 callers
Method
name
imperative/src/impl/ops/custom_opdef.cpp:41
↓ 25 callers
Method
name
! * \brief name of this data type */
dnn/include/megdnn/dtype.h:448
↓ 25 callers
Method
remove_axis
dnn/src/common/basic_types.cpp:245
↓ 25 callers
Function
run_test
src/opr/test/internal.cpp:53
↓ 25 callers
Method
set
dnn/src/rocm/convolution/helper.cpp:31
↓ 25 callers
Method
set_data_by_share
share the data to the tensor Args: data: the data will shared to the tensor, it should be a numpy.ndarra
lite/pylite/megenginelite/tensor.py:420
↓ 25 callers
Method
sync
imperative/src/impl/persistent_cache.cpp:121
↓ 25 callers
Method
tensors
r""" Returns an iterable for the :class:`~.Tensor` of the module. :param recursive: If ``True``, returns all :class:`~.Tensor` within
imperative/python/megengine/module/module.py:300
↓ 25 callers
Function
update
dnn/src/naive/lamb/opr_impl.cpp:15
↓ 24 callers
Function
STBI__CASE
lite/example/cpp_example/mge/cv/stb_image.h:5140
↓ 24 callers
Function
accumulate_2_q_vector
dnn/src/arm_common/conv_bias/int8/channel_wise_kernel.cpp:11
↓ 24 callers
Function
benchmark_impl
dnn/test/x86/conv_bias.cpp:2216
↓ 24 callers
Method
detach
r"""Returns a new :class:`~.Tensor`, detached from the current graph.
imperative/python/megengine/tensor.py:157
↓ 24 callers
Method
dtype
imperative/python/src/tensor.cpp:684
↓ 24 callers
Function
enable_miopen_algo_search
dnn/src/rocm/utils.h:24
↓ 24 callers
Method
exp
dnn/include/megdnn/dtype/half.hpp:1193
↓ 24 callers
Method
get
src/tensorrt/impl/tensorrt_engine_cache.cpp:198
↓ 24 callers
Function
get_cv_args
dnn/test/common/resize.h:156
↓ 24 callers
Function
get_default_device
r"""Gets default computing node. It returns the value set by :func:`~.set_default_device`.
imperative/python/megengine/device.py:147
↓ 24 callers
Method
insert
(self, idx, mod: Module)
imperative/python/megengine/traced_module/utils.py:242
↓ 24 callers
Function
isfinite
Check if finite number. \param arg number to check \retval true if neither infinity nor NaN \retval false else template<typename T> typename enable<bo
dnn/include/megdnn/dtype/half.hpp:2640
↓ 24 callers
Method
proxy_to_default_cpu
src/core/include/megbrain/tensor.h:545
↓ 24 callers
Function
py_apply
imperative/python/src/tensor.cpp:82
↓ 24 callers
Function
rep
dnn/test/common/conv_bias.cpp:1166
↓ 24 callers
Method
reset
imperative/src/impl/proxy_graph/mini_graph.h:138
↓ 24 callers
Function
sc
dnn/src/cuda/cudnn_wrapper.cpp:329
↓ 24 callers
Method
shape
imperative/python/src/tensor.cpp:671
↓ 24 callers
Function
tester
(x_shape, normalized_shape, w_shape, b_shape, eps=1e-5, dtype=None)
imperative/python/test/unit/xla/functional/test_xla_normalize.py:19
↓ 24 callers
Function
var_from_tensor_shape
! * \brief get a symbolvar whose value is tensor shape * * \param inp used to determine the computing graph, which can be any symbolvar * bel
src/core/include/megbrain/graph/helper.h:254
↓ 24 callers
Method
wrap_outputs
imperative/src/impl/transformations/format.cpp:91
↓ 23 callers
Method
Elemwise
src/opr/impl/basic_arith.cpp:66
↓ 23 callers
Function
apply
imperative/src/include/megbrain/imperative/dispatch.h:40
↓ 23 callers
Method
args
r"""Get the positional arguments of the operation corresponding to this Expr.
imperative/python/megengine/traced_module/expr.py:222
↓ 23 callers
Function
check_winograd
dnn/test/common/conv_bias.cpp:1215
↓ 23 callers
Function
convert_to_cnnl_datatype
dnn/src/cambricon/cnnl_wrapper/cnnl_types.cpp:7
↓ 23 callers
Method
get_input_layout_constraint
imperative/src/impl/op_def.cpp:52
↓ 23 callers
Function
interleave_4
dnn/src/armv7/matrix_mul/asm/common.h:618
↓ 23 callers
Method
is_empty
dnn/src/common/basic_types.cpp:177
↓ 23 callers
Function
megcoreCreateDeviceHandle
dnn/src/common/megcore/public_api/device.cpp:9
↓ 23 callers
Function
megcoreMemcpy
dnn/src/common/megcore/public_api/computing.cpp:39
↓ 23 callers
Function
modify_opr_algo_strategy_inplace
r"""C++ graph version of :func:`~.set_execution_strategy`. Used to inplacely modify dumped graph's fast-run strategy. Args: dest_vars
imperative/python/megengine/core/tensor/megbrain_graph.py:343
↓ 23 callers
Function
mul
(inputs, f, c)
imperative/python/test/unit/core/test_subgraph.py:129
↓ 23 callers
Method
set_data_by_copy
copy the data to the tensor Args: data: the data to copy to tensor, it should be list, numpy.ndarraya or
lite/pylite/megenginelite/tensor.py:453
↓ 22 callers
Function
_mm_cvtepi8_epi16_from_ptr
dnn/src/x86/matrix_mul/common/common.h:537
↓ 22 callers
Method
as_ref
imperative/src/include/megbrain/imperative/value.h:203
↓ 22 callers
Function
cnnl_handle
dnn/src/cambricon/utils.h:21
↓ 22 callers
Function
get_conv_bias_args
dnn/test/common/conv_bias.cpp:1272
↓ 22 callers
Method
get_single_comp_node
src/core/impl/graph/operator_node.cpp:487
↓ 22 callers
Method
get_workspace_size
dnn/src/aarch64/conv_bias/int8/strategy.cpp:210
↓ 22 callers
Method
graph_compile
src/serialization/impl/serializer.cpp:29
↓ 22 callers
Function
interleave_helper
dnn/src/armv7/matrix_mul/asm/common.h:586
↓ 22 callers
Function
is_profiling
()
imperative/python/megengine/utils/profiler.py:322
↓ 22 callers
Method
locator
get the physical locator that created this comp node
src/core/include/megbrain/comp_node.h:445
↓ 22 callers
Function
make
imperative/src/include/megbrain/imperative/opr_utility.h:60
↓ 22 callers
Function
merge_one_out
src/opr/test/cond.cpp:45
↓ 22 callers
Function
prefetch_3x
dnn/src/armv7/matrix_mul/asm/common.h:54
↓ 22 callers
Method
reset
imperative/src/impl/proxy_graph.cpp:431
↓ 22 callers
Method
reset
src/core/impl/graph/var_node.cpp:13
↓ 22 callers
Function
setup_algo
imperative/src/impl/algo_chooser.h:10
↓ 22 callers
Method
usable_output
src/core/impl/graph/operator_node.cpp:167
↓ 21 callers
Function
async_error_info
dnn/src/cuda/utils.h:33
↓ 21 callers
Function
cb1
(_, grad)
imperative/python/test/unit/xla/test_xla_training.py:312
↓ 21 callers
Function
check_compute_capability
dnn/test/cuda/utils.cpp:5
↓ 21 callers
Function
checker_conv_bias_qint8x8x8
dnn/test/arm_common/conv_bias_multi_thread.cpp:290
↓ 21 callers
Method
comp_node
src/tensorrt/include/megbrain/tensorrt/tensorrt_opr.h:223
↓ 21 callers
Method
dequantize
dnn/include/megdnn/dtype.h:854
↓ 21 callers
Method
digest
src/core/impl/utils/hash.cpp:22
↓ 21 callers
Method
exec
dnn/src/common/powc.cpp:8
↓ 21 callers
Method
get
imperative/src/include/megbrain/imperative/utils/local_ptr.h:210
↓ 21 callers
Function
get_conv_bias_1x1_args
dnn/test/common/conv_bias.cpp:1356
↓ 21 callers
Method
get_scale
dnn/src/common/utils.cpp:239
↓ 21 callers
Method
handle
imperative/src/include/megbrain/imperative/transformations/eval.h:25
↓ 21 callers
Method
impl
src/core/include/megbrain/utils/thin/hash_table.h:314
↓ 21 callers
Function
init_output_static_infer_desc
src/opr/impl/loop/impl.cpp:163
↓ 21 callers
Function
interleave_4
dnn/src/aarch64/matrix_mul/asm/common.h:1293
↓ 21 callers
Method
load
src/opr/impl/dnn/dnn.sereg.h:212
↓ 21 callers
Method
lock
imperative/src/impl/value.cpp:145
↓ 21 callers
Function
megcoreGetDeviceHandle
dnn/src/common/megcore/public_api/computing.cpp:25
↓ 21 callers
Method
offset
imperative/src/include/megbrain/imperative/physical_tensor.h:140
↓ 21 callers
Method
quantize
dnn/include/megdnn/dtype.h:845
↓ 21 callers
Method
select
src/jit/impl/mlir/ir/common.cpp:98
↓ 21 callers
Function
set_symbolic_shape
r"""Set the VarNode use symbolic shape or not, return the last status. Please set to True and must recover after dump if want to change the input
imperative/python/megengine/utils/network.py:535
↓ 21 callers
Function
tensor_formats_to_named_tensor_shape
src/gopt/impl/global_layout_transform/utils.h:97
← previous
next →
601–700 of 33,912, ranked by callers