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
↓ 13 callers
Function
tester
(ishapes, dtype=None)
imperative/python/test/unit/xla/functional/test_xla_communicate.py:22
↓ 13 callers
Method
update
update the member from C, this will auto used after slice, share
lite/pylite/megenginelite/tensor.py:343
↓ 13 callers
Function
vmax
dnn/src/arm_common/pooling/pooling_helper.h:714
↓ 13 callers
Function
vset
dnn/src/arm_common/pooling/pooling_helper.h:717
↓ 12 callers
Function
U
dnn/src/naive/rng/opr_impl.cpp:30
↓ 12 callers
Function
_
()
imperative/python/test/unit/core/test_autodiff.py:195
↓ 12 callers
Function
accumulate_2_d_vector
dnn/src/arm_common/conv_bias/int8/channel_wise_kernel.cpp:34
↓ 12 callers
Method
add_layout_constraint
src/core/impl/graph/var_node_mem_mgr.cpp:1233
↓ 12 callers
Method
add_outputs
(self, outputs)
imperative/python/megengine/traced_module/expr.py:133
↓ 12 callers
Method
allocated
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/pushdown.cpp:80
↓ 12 callers
Function
broadcast_to
r"""Broadcasts a tensor to given shape. Args: inp: input tensor. shape: target shape. Returns: output tensor. E
imperative/python/megengine/functional/tensor.py:543
↓ 12 callers
Method
comp_graph
get the associated ComputingGraph
src/gopt/include/megbrain/gopt/framework.h:50
↓ 12 callers
Method
comp_node_valid
! * \brief whether underlying comp_node is valid */
src/core/include/megbrain/tensor.h:181
↓ 12 callers
Function
create_qparams
r""" Args: mode: QuantMode: dtype_meta: Union[str: QuantDtypeMeta]: scale: Tensor: zero_point: Tensor:
imperative/python/megengine/quantization/utils.py:145
↓ 12 callers
Method
create_unique_name
(self, name: str, node: Any = None)
imperative/python/megengine/traced_module/traced_module.py:403
↓ 12 callers
Method
deduce_layout
dnn/src/common/multi_head_attn/proxy_forward_base.cpp:379
↓ 12 callers
Method
dispatch_kern
dnn/src/fallback/convolution/algos.cpp:195
↓ 12 callers
Method
dst
dnn/src/fallback/conv_bias/opr_impl.cpp:814
↓ 12 callers
Method
dtype
imperative/src/include/megbrain/imperative/transformations/complex.h:36
↓ 12 callers
Method
elapsed_time_until
src/core/impl/comp_node/impl_helper.cpp:69
↓ 12 callers
Method
ensure_seed
dnn/src/cuda/rng/opr_impl.h:96
↓ 12 callers
Function
equal
r"""Element-wise equality comparison. Computes the truth value of :math:`x_i == y_i` for each element :math:`x_i` of the input tensor :math:`x`
imperative/python/megengine/functional/elemwise.py:1836
↓ 12 callers
Method
exec
imperative/src/impl/dnn_op_helper.h:245
↓ 12 callers
Method
finished
src/core/impl/comp_node/impl_helper.cpp:13
↓ 12 callers
Function
func
(inp, dout)
imperative/python/test/unit/xla/functional/test_xla_indexing.py:26
↓ 12 callers
Method
get
(self, block=True, timeout=None)
imperative/python/megengine/data/tools/_queue.py:158
↓ 12 callers
Method
get
dnn/src/common/cv/interp_helper.h:150
↓ 12 callers
Function
get_execution_strategy
r"""Returns the execution strategy of :class:`~.module.Conv2d` and :func:`~.matmul` See :func:`~.set_execution_strategy` for possible return valu
imperative/python/megengine/functional/debug_param.py:24
↓ 12 callers
Function
is_const
()
imperative/python/megengine/traced_module/_passes/pattern.py:235
↓ 12 callers
Method
load
(cls, opr)
imperative/python/megengine/utils/network_node.py:648
↓ 12 callers
Function
load_library
dnn/atlas-stub/src/libacl_cblas-wrap.h:143
↓ 12 callers
Method
load_state_dict
r"""Loads a given dictionary created by :func:`state_dict` into this module. If ``strict`` is ``True``, the keys of :func:`state_dict` must ex
imperative/python/megengine/module/module.py:495
↓ 12 callers
Method
locator
src/core/impl/comp_node/rocm/comp_node.cpp:196
↓ 12 callers
Method
low_bit
dnn/include/megdnn/dtype.h:471
↓ 12 callers
Function
make_call_rec
src/opr/test/cond.cpp:38
↓ 12 callers
Function
megcoreDestroyDeviceHandle
dnn/src/common/megcore/public_api/device.cpp:19
↓ 12 callers
Method
multinomial
r"""Random variable with multinomial distribution :math:`\operatorname{Mulitnomial}(p)`. The corresponding probability mass function is
imperative/python/megengine/random/rng.py:537
↓ 12 callers
Function
name
dnn/src/cuda/convolution3d/forward/algo.h:97
↓ 12 callers
Method
name
dnn/src/fallback/conv_bias/algos.h:15
↓ 12 callers
Method
neg
src/gopt/impl/basic_arith/chain.cpp:230
↓ 12 callers
Method
nr_devices
dnn/test/rocm/megcore/fixture.h:9
↓ 12 callers
Method
opr
imperative/src/impl/proxy_graph/mini_graph.h:409
↓ 12 callers
Method
param
dnn/src/cuda/convolution/backward_filter/algo.h:128
↓ 12 callers
Function
powc
src/gopt/test/basic_arith.cpp:21
↓ 12 callers
Method
rank
(self)
imperative/python/megengine/distributed/group.py:73
↓ 12 callers
Method
recorder
get the output recorder if this output var is enabled, or a dummmy recorder otherwise
src/opr/impl/loop/impl.h:50
↓ 12 callers
Function
register_format_rule
imperative/src/impl/transformations/format.cpp:157
↓ 12 callers
Function
rep
dnn/test/x86/elemwise.cpp:17
↓ 12 callers
Method
run
dnn/test/common/svd.cpp:98
↓ 12 callers
Function
run_test
(func, args, ref_shape, is_trace, sym=False)
imperative/python/test/unit/functional/test_elemwise.py:297
↓ 12 callers
Method
set
src/core/include/megbrain/dtype.h:95
↓ 12 callers
Method
shape
imperative/src/impl/proxy_graph/mini_graph.h:190
↓ 12 callers
Function
sigmoid
r"""Element-wise `1 / ( 1 + exp( -x ) )`.
imperative/python/megengine/functional/nn.py:851
↓ 12 callers
Function
sqrt
r"""Element-wise :math:`\operatorname{sqrt}(x)` function. Calculates the square root for each element :math:`x_i` of the input tensor :math:`
imperative/python/megengine/functional/elemwise.py:762
↓ 12 callers
Method
src
dnn/src/fallback/conv_bias/opr_impl.cpp:695
↓ 12 callers
Function
stbi__grow_buffer_unsafe
lite/example/cpp_example/mge/cv/stb_image.h:2218
↓ 12 callers
Function
stbi__zreceive
lite/example/cpp_example/mge/cv/stb_image.h:4543
↓ 12 callers
Method
sub
imperative/src/impl/physical_tensor.cpp:753
↓ 12 callers
Method
tag
! * \brief get corresponding tag for this tag handler */
src/core/impl/graph/static_infer_impl.h:151
↓ 12 callers
Method
to_string
=================== ReformatManager::ReformatKey ====================*/
src/gopt/impl/global_layout_transform/reformat_manager.cpp:67
↓ 12 callers
Function
transpose
dnn/src/x86/utils.cpp:183
↓ 12 callers
Function
unpack_getitem
(inp, tuple_val, *, allow_newaxis=True)
imperative/python/test/unit/core/test_indexing_op.py:136
↓ 12 callers
Method
update_on_opr_auto_replace
src/gopt/impl/framework.cpp:219
↓ 12 callers
Function
worker
(data, expect)
imperative/python/test/unit/functional/test_functional_distributed.py:30
↓ 12 callers
Method
wrap_safe
(cls, value, node)
imperative/python/megengine/traced_module/node.py:362
↓ 12 callers
Method
write_pod
imperative/src/include/megbrain/imperative/ops/opr_attr.h:17
↓ 11 callers
Function
LITE_get_tensor_layout
lite/lite-c/src/tensor.cpp:193
↓ 11 callers
Function
MGB_DEFINE_CLS_WITH_SUPER
src/opr/include/megbrain/opr/internal/megdnn_opr_wrapper.h:245
↓ 11 callers
Method
_dev_tensor
imperative/python/src/tensor.cpp:734
↓ 11 callers
Function
_exclude_from_trace
()
imperative/python/megengine/traced_module/tm_config.py:49
↓ 11 callers
Method
add_axis_cont_inplace
! * \brief add an axis before given *axis*, with shape 1 and contiguous * stride */
dnn/include/megdnn/basic_types.h:288
↓ 11 callers
Method
alloc
src/core/impl/utils/mempool.cpp:19
↓ 11 callers
Method
apply
dnn/src/fallback/powc/opr_impl.cpp:58
↓ 11 callers
Function
apply_on_var_node
imperative/src/impl/ops/rnn.cpp:9
↓ 11 callers
Method
apply_quant_weight
r"""Apply weight's observer and fake_quant from ``qconfig`` on ``target``.
imperative/python/megengine/module/qat/module.py:89
↓ 11 callers
Function
as_device
(obj)
imperative/python/megengine/core/_wrap.py:46
↓ 11 callers
Method
close
(self)
imperative/python/megengine/data/tools/_queue.py:183
↓ 11 callers
Function
collective_comm
r"""Helper function for applying collective communication functions.
imperative/python/megengine/distributed/functional.py:44
↓ 11 callers
Method
cols
dnn/src/common/winograd/winograd_generator.h:83
↓ 11 callers
Method
compile_multi_part
src/core/impl/graph/cg_impl.cpp:455
↓ 11 callers
Method
context
imperative/src/include/megbrain/imperative/cpp_cupti.h:71
↓ 11 callers
Method
define_enum_state
Set up thread-local state and return a contextmanager for managing it. Args: name: string, converted to lowercase to define the name o
imperative/python/megengine/xla/lib/config.py:226
↓ 11 callers
Method
device
imperative/src/impl/value.cpp:48
↓ 11 callers
Method
dump_tensor
src/core/impl/utils/debug.cpp:436
↓ 11 callers
Method
emplace
imperative/src/include/megbrain/imperative/utils/local_ptr.h:95
↓ 11 callers
Function
find
src/core/include/megbrain/utils/metahelper.h:279
↓ 11 callers
Method
find_op
dnn/src/cuda/cutlass/operation_table.cpp:154
↓ 11 callers
Function
fmod
Remainder of division. \param x first operand \param y second operand \return remainder of floating point division. template<typename T,typename U> ty
dnn/include/megdnn/dtype/half.hpp:2162
↓ 11 callers
Function
fn
(shape)
imperative/python/test/unit/random/test_rng.py:806
↓ 11 callers
Function
fwd
(x)
imperative/python/test/unit/utils/test_network.py:21
↓ 11 callers
Method
gen
dnn/test/cuda/argsort.cpp:72
↓ 11 callers
Function
gen_rand_data
src/cambricon/test/magicmind_runtime_opr.cpp:22
↓ 11 callers
Function
generate_transpose_args
dnn/test/common/relayout.cpp:14
↓ 11 callers
Method
get_input_name
get the input name by the index in the network Args: index: the index of the input name Returns: th
lite/pylite/megenginelite/network.py:788
↓ 11 callers
Function
get_shape
imperative/python/src/grad_override.cpp:58
↓ 11 callers
Method
get_type
src/jit/impl/mlir/mlir_gen.cpp:211
↓ 11 callers
Function
get_var_value
(x)
imperative/python/test/helpers/utils.py:31
↓ 11 callers
Function
get_winograd_benchmark_args
dnn/test/x86/conv_bias.cpp:3021
↓ 11 callers
Function
handle_bias_and_nonlinear
dnn/src/common/conv_bias.cpp:281
↓ 11 callers
Method
hash
imperative/src/impl/ops/opr_attr.cpp:166
← previous
next →
1,001–1,100 of 33,912, ranked by callers