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
↓ 46 callers
Function
add
r"""Element-wise addition. Calculates the sum for each element :math:`x_i` of the input tensor :math:`x` with the respective element :math:`y
imperative/python/megengine/functional/elemwise.py:76
↓ 46 callers
Method
find
dnn/include/megdnn/thin/small_vector.h:160
↓ 46 callers
Method
infer_value
imperative/src/impl/proxy_graph/mini_graph.h:520
↓ 46 callers
Method
megcore_computing_handle
! * \brief Get the underlying megcore computing handle. */
dnn/include/megdnn/handle.h:107
↓ 46 callers
Method
name
dnn/src/naive/matrix_mul/algorithms.h:9
↓ 45 callers
Function
_mm256_loadu2_m128_emulate
dnn/src/x86/avx_helper.h:20
↓ 45 callers
Function
powf
src/opr/include/megbrain/opr/basic_arith_wrapper.h:71
↓ 45 callers
Method
read
dnn/src/common/cv/mat.cpp:129
↓ 45 callers
Function
tester
(x_shape, y_shape, indices, dtype=None)
imperative/python/test/unit/xla/functional/test_xla_indexing.py:104
↓ 45 callers
Method
type
src/custom/impl/param.cpp:71
↓ 44 callers
Method
flatten
r"""Get a new TracedModule, which eliminates ``GetAttr`` and has no hierarchy. Retruns: A new :class:`TracedModule`.
imperative/python/megengine/traced_module/traced_module.py:2159
↓ 44 callers
Function
get_workspace
dnn/src/x86/matrix_mul/algos.h:13
↓ 44 callers
Function
get_workspace_in_bytes
dnn/src/naive/rnn/funcs.h:23
↓ 44 callers
Function
is_static_var_shape
! * \brief whether var shape could be statically inferred */
src/core/include/megbrain/graph/helper.h:22
↓ 44 callers
Method
max
(self, axis, keepdims=False)
imperative/python/megengine/xla/rules/hlotensor.py:186
↓ 44 callers
Method
numpy
imperative/python/src/tensor.h:73
↓ 44 callers
Function
opr_test
:param cases: the list which have dict element, the list length should be 2 for dynamic shape test. and the dict should have input,
imperative/python/test/helpers/utils.py:40
↓ 44 callers
Method
set_value
lite/load_and_run/src/helpers/common.h:157
↓ 44 callers
Function
stbi__rewind
lite/example/cpp_example/mge/cv/stb_image.h:900
↓ 44 callers
Method
transpose
(self, permutation)
imperative/python/megengine/xla/rules/hlotensor.py:151
↓ 43 callers
Method
clear_grad
(self)
imperative/python/test/unit/jit/test_tracing.py:873
↓ 43 callers
Method
dtype
(self)
imperative/python/megengine/xla/rules/hlotensor.py:52
↓ 43 callers
Function
log
Natural logorithm. \param arg function argument \return logarithm of \a arg to base e template<typename T> typename enable<expr,T>::type log(T arg) {
dnn/include/megdnn/dtype/half.hpp:2267
↓ 43 callers
Method
make_rewriter
src/gopt/include/megbrain/gopt/framework.h:129
↓ 42 callers
Function
GiSt1Float32
dnn/src/fallback/general_intrinsic/gi_float.h:366
↓ 42 callers
Method
exec
dnn/src/naive/rnn/opr_impl.cpp:23
↓ 42 callers
Method
get_secs
src/core/include/megbrain/utils/timer.h:33
↓ 42 callers
Method
make_scalar_dt
src/core/include/megbrain/graph/symbol_var.h:112
↓ 42 callers
Method
mul
dnn/src/common/winograd/winograd_generator.cpp:179
↓ 42 callers
Method
replace
(self, **kw)
imperative/python/megengine/xla/lower.py:70
↓ 42 callers
Method
step
(self)
imperative/python/megengine/utils/profiler.py:198
↓ 42 callers
Function
zeros
(shape, dtype=np.float32)
imperative/python/megengine/xla/rules/tensor.py:203
↓ 41 callers
Method
all
(self, axis, keepdims=False)
imperative/python/megengine/xla/rules/hlotensor.py:196
↓ 41 callers
Method
dist_elem
dnn/include/megdnn/basic_types.h:152
↓ 41 callers
Method
info
imperative/src/impl/profiler/states.h:38
↓ 41 callers
Function
make_call
src/jit/include/megbrain/jit/ast_c.h:189
↓ 41 callers
Method
output_dtype
src/core/include/megbrain/graph/operator_node.h:115
↓ 41 callers
Function
sum
r"""Calculates the sum of tensor elements over a given axis (or axes). Args: inp: input tensor. Should have a numeric data type.
imperative/python/megengine/functional/math.py:150
↓ 41 callers
Method
symshape
src/core/impl/graph/symbol_var.cpp:17
↓ 41 callers
Method
to_numpy
get the buffer of the tensor
lite/pylite/megenginelite/tensor.py:515
↓ 41 callers
Method
total_nr_elems
imperative/src/include/megbrain/imperative/utils/value_shape.h:38
↓ 41 callers
Function
try_cast
lite/include/lite/tensor.h:440
↓ 40 callers
Method
collapse_contiguous
* \brief The collapse_contiguous function will convert a contiguous image like * tensor layout into a 2-dimensional layout, shape[0] = height of the
dnn/src/common/basic_types.cpp:334
↓ 40 callers
Function
get_ABC
dnn/src/common/reduce_helper.cpp:10
↓ 40 callers
Function
get_device_count
r"""Gets number of devices installed on this system. Args: device_type: device type, one of 'gpu' or 'cpu'
imperative/python/megengine/device.py:89
↓ 40 callers
Method
get_rank
src/opr-mm/include/megbrain/opr/group_manager.h:34
↓ 40 callers
Function
get_workspace_in_bytes
dnn/src/cuda/batched_matrix_mul/algo.h:108
↓ 40 callers
Method
name
src/tensorrt/impl/opr_replace.cpp:1602
↓ 40 callers
Function
prefetch_3x
dnn/src/aarch64/matrix_mul/asm/common.h:57
↓ 40 callers
Method
seed
(self)
imperative/python/megengine/module/vision.py:102
↓ 40 callers
Method
store
dnn/src/arm_common/resize/helper.h:21
↓ 40 callers
Function
trace_module
r"""Traces module ``mod`` and returns corresponding :class:`TracedModule`. Args: mod(:attr:`.module.Module`): the module will be converte
imperative/python/megengine/traced_module/traced_module.py:2442
↓ 39 callers
Function
copy
r"""Copies tensor to another device. Args: inp: input tensor. device: destination device. Examples: >>> import nump
imperative/python/megengine/functional/tensor.py:1209
↓ 39 callers
Function
cudnn_handle
dnn/src/cuda/utils.h:21
↓ 39 callers
Function
mgb_lar
lite/test/test_common.h:117
↓ 39 callers
Function
next
! * \brief get the interval whose address is greater than newly * allocated one, or nullptr if the new interval is at end */
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/pushdown.cpp:170
↓ 39 callers
Method
property
src/jit/impl/mlir/compiler.h:23
↓ 39 callers
Function
stbi__get16be
lite/example/cpp_example/mge/cv/stb_image.h:1732
↓ 39 callers
Method
train
r"""Sets training mode of all the modules within this module (including itself) to ``mode``. This effectively sets the ``training`` attributes
imperative/python/megengine/module/module.py:392
↓ 39 callers
Method
valid
src/gopt/impl/global_layout_transform/dynamic_programming_solver.cpp:30
↓ 38 callers
Method
id
imperative/src/impl/interpreter/stack_manager.h:136
↓ 38 callers
Function
make
src/core/test/sublinear_memory.cpp:71
↓ 38 callers
Method
name
src/serialization/impl/serializer_oss_common.h:44
↓ 38 callers
Method
reset
src/opr/impl/loop/impl.h:417
↓ 38 callers
Method
to_string
src/core/impl/graph/cg_impl.cpp:879
↓ 37 callers
Method
copy
imperative/python/src/tensor.h:51
↓ 37 callers
Function
f
(*args)
imperative/python/test/unit/core/test_autodiff.py:25
↓ 37 callers
Method
get
imperative/src/impl/persistent_cache.cpp:60
↓ 37 callers
Method
get
src/opr/test/dnn/batch_norm.cpp:34
↓ 37 callers
Function
is_op
(target)
imperative/python/megengine/traced_module/_passes/pattern.py:221
↓ 37 callers
Function
megcoreSynchronize
dnn/src/common/megcore/public_api/computing.cpp:54
↓ 36 callers
Method
dtype
(self)
lite/pylite/megenginelite/tensor.py:107
↓ 36 callers
Method
encode
lite/load_and_run/src/helpers/utils.cpp:56
↓ 36 callers
Method
exec
dnn/test/rocm/chanwise_convolution.cpp:121
↓ 36 callers
Function
get_bundle
dnn/src/arm_common/pooling/algo.cpp:18
↓ 36 callers
Function
hash
imperative/src/impl/ops/utility.cpp:224
↓ 36 callers
Method
make
imperative/src/impl/interpreter/stack_manager.h:74
↓ 36 callers
Method
remove
(self)
imperative/python/megengine/utils/hook.py:13
↓ 36 callers
Method
reserve
Used for initialization
dnn/src/cuda/cutlass/manifest.cpp:49
↓ 36 callers
Method
scope
imperative/src/impl/op_def.cpp:131
↓ 35 callers
Method
assert_values_in_order
dnn/test/common/small_vector.cpp:199
↓ 35 callers
Function
f
(x)
imperative/python/test/unit/functional/test_elemwise.py:338
↓ 35 callers
Method
gen_single_val
dnn/test/common/rng.cpp:214
↓ 35 callers
Method
get
imperative/python/src/grad.cpp:139
↓ 35 callers
Method
ptr
src/core/include/megbrain/tensor.h:383
↓ 35 callers
Method
record
src/core/impl/comp_node/impl_helper.cpp:5
↓ 35 callers
Function
tree_flatten
r"""Flattens a pytree into a list of values and a :class:`TreeDef` that can be used to reconstruct the pytree.
imperative/python/megengine/traced_module/pytree.py:199
↓ 35 callers
Method
valid
dnn/test/common/workspace_wrapper.h:16
↓ 34 callers
Function
apply
src/jit/impl/halide/ast_hl.cpp:377
↓ 34 callers
Method
format
imperative/src/impl/value.cpp:61
↓ 34 callers
Method
get_input_tensor
lite/src/network.cpp:139
↓ 34 callers
Function
get_option
imperative/src/impl/interpreter/option_manager.h:55
↓ 34 callers
Method
insert
imperative/src/impl/op_trait.h:223
↓ 34 callers
Method
is_physical_contiguous
dnn/src/common/basic_types.cpp:283
↓ 34 callers
Method
name
imperative/python/src/tensor.h:82
↓ 34 callers
Method
param
dnn/src/fallback/conv_bias/im2col/algos.h:60
↓ 34 callers
Method
prev_dev_ptr
! * \brief previous dev ptr before deallocating dev_tensor; used for * testing and debugging */
src/core/include/megbrain/graph/var_node.h:365
↓ 34 callers
Method
to
r"""Copy self :class:`~.Tensor` to specified device. See :func:`~.copy`
imperative/python/megengine/tensor.py:198
↓ 33 callers
Function
TensorValueLowbit4
dnn/test/common/checker.h:451
↓ 33 callers
Method
erase
src/custom/impl/manager.cpp:122
← previous
next →
401–500 of 33,912, ranked by callers