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
↓ 8 callers
Function
MIDOUT_BEGIN
dnn/src/naive/conv_bias/opr_impl.cpp:216
↓ 8 callers
Function
_Const
imperative/python/src/tensor_utils.cpp:307
↓ 8 callers
Method
__init__
( self, in_channels: int, out_channels: int, kernel_size: int, stride:
imperative/python/megengine/module/conv.py:163
↓ 8 callers
Method
__init__
(self, name, qualname)
imperative/python/megengine/traced_module/traced_module.py:398
↓ 8 callers
Method
__init__
(self, loader, preload)
imperative/python/megengine/data/dataloader.py:249
↓ 8 callers
Function
_get_expr_checker
()
imperative/python/megengine/traced_module/tm_config.py:13
↓ 8 callers
Method
_get_image
(self, input: Tuple)
imperative/python/megengine/data/transform/vision/transform.py:117
↓ 8 callers
Function
_make_shape_tuple
imperative/python/src/tensor_utils.cpp:331
↓ 8 callers
Method
activate
src/core/impl/comp_node/rocm/comp_node.cpp:119
↓ 8 callers
Method
activate
src/core/impl/comp_node/cambricon/comp_node.cpp:112
↓ 8 callers
Method
add_dest
src/core/impl/graph/static_infer_impl.cpp:1033
↓ 8 callers
Function
align_size
dnn/src/x86/separable_conv/sep_conv_common.h:79
↓ 8 callers
Method
alloc_unique
src/core/include/megbrain/utils/mempool.h:83
↓ 8 callers
Method
apply_quant_bias
r"""Use :func:`~.fake_quant_bias` to process ``target``. Only valid when ``act_fake_quant`` and ``weight_fake_quant`` are both enabled.
imperative/python/megengine/module/qat/module.py:101
↓ 8 callers
Method
astype
r"""Returns a :class:`Tensor` with the same data and number of elements with the specified :attr:`~.Tensor.dtype`.
imperative/python/megengine/core/tensor/array_method.py:390
↓ 8 callers
Method
backend
(self)
imperative/python/megengine/xla/lower.py:65
↓ 8 callers
Method
backward
r"""Compute gradients (or vector-Jacobian product) for all attached tensors, accumulate to corresponding .grad attribute, and release resource
imperative/python/megengine/autodiff/grad_manager.py:229
↓ 8 callers
Method
bcast_val
(self, val, key, size)
imperative/python/megengine/distributed/server.py:294
↓ 8 callers
Function
benchmark_nchw44_8x8x16_vs_8x8x32
dnn/test/arm_common/conv_bias.cpp:536
↓ 8 callers
Function
bool_env
Read an environment variable and interpret it as a boolean. True values are (case insensitive): 'y', 'yes', 't', 'true', 'on', and '1'; false
imperative/python/megengine/xla/lib/config.py:24
↓ 8 callers
Function
build_dtype
src/serialization/impl/flatbuffers_helper.cpp:68
↓ 8 callers
Function
builtin_signbit
dnn/include/megdnn/dtype/half.hpp:293
↓ 8 callers
Function
char_length
lite/load_and_run/src/helpers/text_table.cpp:13
↓ 8 callers
Function
check_dtype
dnn/src/cambricon/argsort/opr_impl.cpp:14
↓ 8 callers
Function
check_param
(param, info)
imperative/python/megengine/xla/rules/nn.py:373
↓ 8 callers
Function
checker_conv_bias_common
dnn/test/common/conv_bias.cpp:1484
↓ 8 callers
Function
checker_conv_bias_preprocess
dnn/test/x86/conv_bias.cpp:1458
↓ 8 callers
Method
clear
imperative/python/src/external_convert.h:72
↓ 8 callers
Function
deprecated_func
r""" Args: version: version to deprecate this function origin: origin module path name: function name tbd: to be
imperative/python/megengine/utils/deprecation.py:10
↓ 8 callers
Method
dev_tensor
get device tensor
lite/src/mge/tensor_impl.h:91
↓ 8 callers
Function
do_exec
dnn/src/cuda/mesh_indexing/opr_impl.cpp:28
↓ 8 callers
Method
empty
imperative/src/include/megbrain/imperative/subgraph.h:79
↓ 8 callers
Function
enable_fastpath
imperative/python/src/tensor_utils.cpp:927
↓ 8 callers
Method
enable_miopen_algo_search
dnn/src/rocm/handle.h:35
↓ 8 callers
Function
enabled
imperative/src/impl/cpp_cupti.cpp:223
↓ 8 callers
Method
endpoint_contain
whether endpoint contain a given var node
src/gopt/include/megbrain/gopt/framework.h:63
↓ 8 callers
Method
exit
imperative/python/src/grad.cpp:124
↓ 8 callers
Function
expand_dims
(inp, axis)
imperative/python/megengine/xla/rules/tensor.py:81
↓ 8 callers
Function
fake_quant_tensor
Apply fake quantization to the inp tensor. Args: inp: the input tensor which need to be faked. qparams: to get mode, qmin, qmax,
imperative/python/megengine/quantization/utils.py:170
↓ 8 callers
Function
fill_zero_dev_tensor
! * \brief fill zeros in the content of a dev tensor */
src/core/include/megbrain/tensor.h:572
↓ 8 callers
Function
find_index
imperative/src/impl/proxy_graph/mini_graph.h:20
↓ 8 callers
Method
get
src/core/impl/utils/infile_persistent_cache.cpp:239
↓ 8 callers
Method
get
dnn/src/fallback/conv_bias/conv1x1/conv1x1_utils.h:52
↓ 8 callers
Method
get_algorithm_from_desc
dnn/src/naive/pooling/opr_impl.cpp:572
↓ 8 callers
Method
get_all_input_name
lite/src/network.cpp:241
↓ 8 callers
Function
get_int8_chwn4_args
dnn/test/common/conv_bias.cpp:676
↓ 8 callers
Function
get_int8_nchw44_args
dnn/test/common/conv_bias.cpp:404
↓ 8 callers
Function
get_int8_nchw4_args_check_bounds
dnn/test/common/conv_bias.cpp:492
↓ 8 callers
Function
get_int8_quint8_conv_bias_args
dnn/test/arm_common/conv_bias_multi_thread.cpp:12
↓ 8 callers
Method
get_msecs
dnn/test/common/timer.h:58
↓ 8 callers
Function
get_offsets
(shapes)
imperative/python/megengine/distributed/helper.py:98
↓ 8 callers
Function
get_promoted_dtype
imperative/src/impl/transformations/dtype_promote.cpp:45
↓ 8 callers
Function
get_value_dtypes
imperative/src/impl/transformations/dtype_promote.cpp:37
↓ 8 callers
Method
get_wrapped_type
(cls, value)
imperative/python/megengine/traced_module/node.py:385
↓ 8 callers
Method
impl
dnn/src/arm_common/neon_struct.h:32
↓ 8 callers
Method
infer
src/core/impl/graph/static_infer_impl.cpp:383
↓ 8 callers
Function
init
src/core/impl/graph/static_infer_impl.cpp:299
↓ 8 callers
Method
init_desc
dnn/src/cuda/convolution3d/forward/algo.h:42
↓ 8 callers
Function
interpolate_cubic
dnn/src/x86/resize/resize_cv.cpp:1405
↓ 8 callers
Function
is_bool_dtype
imperative/python/src/tensor_utils.cpp:274
↓ 8 callers
Function
is_call_function
(expr, func: Iterable[Callable] = None)
imperative/python/megengine/traced_module/expr.py:65
↓ 8 callers
Function
is_constant
(expr)
imperative/python/megengine/traced_module/expr.py:73
↓ 8 callers
Method
is_continuous
dnn/src/common/cv/mat.cpp:124
↓ 8 callers
Method
isfinite
dnn/include/megdnn/dtype/half.hpp:1790
↓ 8 callers
Function
make_deconv
user interface for deconv
src/opr/include/megbrain/opr/dnn/convolution.h:139
↓ 8 callers
Function
matmul_deduce_layout
dnn/src/common/multi_head_attn/helper.h:14
↓ 8 callers
Function
matmul_exec
dnn/src/common/multi_head_attn/helper.h:34
↓ 8 callers
Method
max
dnn/test/common/rng.cpp:23
↓ 8 callers
Function
mbedtls_aes_crypt_ecb
* AES-ECB block encryption/decryption */
lite/src/decryption/mbedtls/aes.cc:818
↓ 8 callers
Function
megcoreActivate
dnn/src/common/megcore/public_api/device.cpp:52
↓ 8 callers
Method
min
dnn/test/common/rng.cpp:22
↓ 8 callers
Method
name
imperative/src/include/megbrain/imperative/transformations/grad.h:241
↓ 8 callers
Method
nr_alloc
src/core/test/mem_alloc.cpp:79
↓ 8 callers
Method
offset
! * \brief offset on allocated block in bytes */
src/core/include/megbrain/tensor.h:167
↓ 8 callers
Function
on
src/jit/impl/halide/ast_hl.cpp:380
↓ 8 callers
Method
opr
get the opr impl so setting other than param() can be modified
dnn/test/common/checker.h:315
↓ 8 callers
Function
polynomial
src/jit/impl/mlir/ir/numerical.cpp:9
↓ 8 callers
Method
print_used_time
lite/example/cpp_example/helper.h:74
↓ 8 callers
Method
procedural_name
(self)
dnn/scripts/cutlass_generator/library.py:677
↓ 8 callers
Method
put
imperative/python/src/utils.cpp:439
↓ 8 callers
Method
put
imperative/src/impl/persistent_cache.cpp:82
↓ 8 callers
Method
realpath
src/jit/impl/utils.cpp:210
↓ 8 callers
Method
record
record an event on cn
imperative/src/impl/physical_tensor.cpp:320
↓ 8 callers
Method
rend
dnn/include/megdnn/thin/small_vector.h:148
↓ 8 callers
Method
replace_vars
r"""Replaces vars in the graph. Args: repl_dict: the map {old_var: new_var} that specifies how to replace the vars.
imperative/python/megengine/utils/network.py:378
↓ 8 callers
Function
repr_pyobj
imperative/python/src/helper.cpp:66
↓ 8 callers
Function
run_all_gather
src/opr/test/muxing.cpp:14
↓ 8 callers
Function
safe_call
(func, qparams)
imperative/python/megengine/quantization/quantize.py:150
↓ 8 callers
Method
set
dnn/src/rocm/convolution/helper.h:66
↓ 8 callers
Method
set_log_level
imperative/python/src/utils.cpp:47
↓ 8 callers
Function
set_symbolic_shape
r"""Sets whether tensor.shape returns a tensor instead of a tuple
imperative/python/megengine/core/_trace_option.py:19
↓ 8 callers
Method
shape
imperative/src/include/megbrain/imperative/basic_operators.h:107
↓ 8 callers
Method
size
src/core/include/megbrain/utils/thin/hash_table.h:357
↓ 8 callers
Method
size_log
size() in log2
dnn/include/megdnn/dtype.h:503
↓ 8 callers
Method
static_device_memory_manager
set underlying device memory manager
src/core/impl/graph/var_node_mem_mgr.h:202
↓ 8 callers
Function
stbi__clamp
take a -128..127 value and stbi__clamp it and convert to 0..255
lite/example/cpp_example/mge/cv/stb_image.h:2564
↓ 8 callers
Function
stbi__convert_format
lite/example/cpp_example/mge/cv/stb_image.h:1792
↓ 8 callers
Function
stbi__mad3sizes_valid
returns 1 if "a*b*c + add" has no negative terms/factors and doesn't overflow
lite/example/cpp_example/mge/cv/stb_image.h:1057
↓ 8 callers
Function
stbir__use_upsampling
lite/example/cpp_example/mge/cv/stb_image_resize.h:887
↓ 8 callers
Function
stbiw__putc
lite/example/cpp_example/mge/cv/stb_image_write.h:423
← previous
next →
1,301–1,400 of 33,912, ranked by callers