MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 8 callersFunctionMIDOUT_BEGIN
dnn/src/naive/conv_bias/opr_impl.cpp:216
↓ 8 callersFunction_Const
imperative/python/src/tensor_utils.cpp:307
↓ 8 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: int, stride:
imperative/python/megengine/module/conv.py:163
↓ 8 callersMethod__init__
(self, name, qualname)
imperative/python/megengine/traced_module/traced_module.py:398
↓ 8 callersMethod__init__
(self, loader, preload)
imperative/python/megengine/data/dataloader.py:249
↓ 8 callersFunction_get_expr_checker
()
imperative/python/megengine/traced_module/tm_config.py:13
↓ 8 callersMethod_get_image
(self, input: Tuple)
imperative/python/megengine/data/transform/vision/transform.py:117
↓ 8 callersFunction_make_shape_tuple
imperative/python/src/tensor_utils.cpp:331
↓ 8 callersMethodactivate
src/core/impl/comp_node/rocm/comp_node.cpp:119
↓ 8 callersMethodactivate
src/core/impl/comp_node/cambricon/comp_node.cpp:112
↓ 8 callersMethodadd_dest
src/core/impl/graph/static_infer_impl.cpp:1033
↓ 8 callersFunctionalign_size
dnn/src/x86/separable_conv/sep_conv_common.h:79
↓ 8 callersMethodalloc_unique
src/core/include/megbrain/utils/mempool.h:83
↓ 8 callersMethodapply_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 callersMethodastype
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 callersMethodbackend
(self)
imperative/python/megengine/xla/lower.py:65
↓ 8 callersMethodbackward
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 callersMethodbcast_val
(self, val, key, size)
imperative/python/megengine/distributed/server.py:294
↓ 8 callersFunctionbenchmark_nchw44_8x8x16_vs_8x8x32
dnn/test/arm_common/conv_bias.cpp:536
↓ 8 callersFunctionbool_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 callersFunctionbuild_dtype
src/serialization/impl/flatbuffers_helper.cpp:68
↓ 8 callersFunctionbuiltin_signbit
dnn/include/megdnn/dtype/half.hpp:293
↓ 8 callersFunctionchar_length
lite/load_and_run/src/helpers/text_table.cpp:13
↓ 8 callersFunctioncheck_dtype
dnn/src/cambricon/argsort/opr_impl.cpp:14
↓ 8 callersFunctioncheck_param
(param, info)
imperative/python/megengine/xla/rules/nn.py:373
↓ 8 callersFunctionchecker_conv_bias_common
dnn/test/common/conv_bias.cpp:1484
↓ 8 callersFunctionchecker_conv_bias_preprocess
dnn/test/x86/conv_bias.cpp:1458
↓ 8 callersMethodclear
imperative/python/src/external_convert.h:72
↓ 8 callersFunctiondeprecated_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 callersMethoddev_tensor
get device tensor
lite/src/mge/tensor_impl.h:91
↓ 8 callersFunctiondo_exec
dnn/src/cuda/mesh_indexing/opr_impl.cpp:28
↓ 8 callersMethodempty
imperative/src/include/megbrain/imperative/subgraph.h:79
↓ 8 callersFunctionenable_fastpath
imperative/python/src/tensor_utils.cpp:927
↓ 8 callersMethodenable_miopen_algo_search
dnn/src/rocm/handle.h:35
↓ 8 callersFunctionenabled
imperative/src/impl/cpp_cupti.cpp:223
↓ 8 callersMethodendpoint_contain
whether endpoint contain a given var node
src/gopt/include/megbrain/gopt/framework.h:63
↓ 8 callersMethodexit
imperative/python/src/grad.cpp:124
↓ 8 callersFunctionexpand_dims
(inp, axis)
imperative/python/megengine/xla/rules/tensor.py:81
↓ 8 callersFunctionfake_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 callersFunctionfill_zero_dev_tensor
! * \brief fill zeros in the content of a dev tensor */
src/core/include/megbrain/tensor.h:572
↓ 8 callersFunctionfind_index
imperative/src/impl/proxy_graph/mini_graph.h:20
↓ 8 callersMethodget
src/core/impl/utils/infile_persistent_cache.cpp:239
↓ 8 callersMethodget
dnn/src/fallback/conv_bias/conv1x1/conv1x1_utils.h:52
↓ 8 callersMethodget_algorithm_from_desc
dnn/src/naive/pooling/opr_impl.cpp:572
↓ 8 callersMethodget_all_input_name
lite/src/network.cpp:241
↓ 8 callersFunctionget_int8_chwn4_args
dnn/test/common/conv_bias.cpp:676
↓ 8 callersFunctionget_int8_nchw44_args
dnn/test/common/conv_bias.cpp:404
↓ 8 callersFunctionget_int8_nchw4_args_check_bounds
dnn/test/common/conv_bias.cpp:492
↓ 8 callersFunctionget_int8_quint8_conv_bias_args
dnn/test/arm_common/conv_bias_multi_thread.cpp:12
↓ 8 callersMethodget_msecs
dnn/test/common/timer.h:58
↓ 8 callersFunctionget_offsets
(shapes)
imperative/python/megengine/distributed/helper.py:98
↓ 8 callersFunctionget_promoted_dtype
imperative/src/impl/transformations/dtype_promote.cpp:45
↓ 8 callersFunctionget_value_dtypes
imperative/src/impl/transformations/dtype_promote.cpp:37
↓ 8 callersMethodget_wrapped_type
(cls, value)
imperative/python/megengine/traced_module/node.py:385
↓ 8 callersMethodimpl
dnn/src/arm_common/neon_struct.h:32
↓ 8 callersMethodinfer
src/core/impl/graph/static_infer_impl.cpp:383
↓ 8 callersFunctioninit
src/core/impl/graph/static_infer_impl.cpp:299
↓ 8 callersMethodinit_desc
dnn/src/cuda/convolution3d/forward/algo.h:42
↓ 8 callersFunctioninterpolate_cubic
dnn/src/x86/resize/resize_cv.cpp:1405
↓ 8 callersFunctionis_bool_dtype
imperative/python/src/tensor_utils.cpp:274
↓ 8 callersFunctionis_call_function
(expr, func: Iterable[Callable] = None)
imperative/python/megengine/traced_module/expr.py:65
↓ 8 callersFunctionis_constant
(expr)
imperative/python/megengine/traced_module/expr.py:73
↓ 8 callersMethodis_continuous
dnn/src/common/cv/mat.cpp:124
↓ 8 callersMethodisfinite
dnn/include/megdnn/dtype/half.hpp:1790
↓ 8 callersFunctionmake_deconv
user interface for deconv
src/opr/include/megbrain/opr/dnn/convolution.h:139
↓ 8 callersFunctionmatmul_deduce_layout
dnn/src/common/multi_head_attn/helper.h:14
↓ 8 callersFunctionmatmul_exec
dnn/src/common/multi_head_attn/helper.h:34
↓ 8 callersMethodmax
dnn/test/common/rng.cpp:23
↓ 8 callersFunctionmbedtls_aes_crypt_ecb
* AES-ECB block encryption/decryption */
lite/src/decryption/mbedtls/aes.cc:818
↓ 8 callersFunctionmegcoreActivate
dnn/src/common/megcore/public_api/device.cpp:52
↓ 8 callersMethodmin
dnn/test/common/rng.cpp:22
↓ 8 callersMethodname
imperative/src/include/megbrain/imperative/transformations/grad.h:241
↓ 8 callersMethodnr_alloc
src/core/test/mem_alloc.cpp:79
↓ 8 callersMethodoffset
! * \brief offset on allocated block in bytes */
src/core/include/megbrain/tensor.h:167
↓ 8 callersFunctionon
src/jit/impl/halide/ast_hl.cpp:380
↓ 8 callersMethodopr
get the opr impl so setting other than param() can be modified
dnn/test/common/checker.h:315
↓ 8 callersFunctionpolynomial
src/jit/impl/mlir/ir/numerical.cpp:9
↓ 8 callersMethodprint_used_time
lite/example/cpp_example/helper.h:74
↓ 8 callersMethodprocedural_name
(self)
dnn/scripts/cutlass_generator/library.py:677
↓ 8 callersMethodput
imperative/python/src/utils.cpp:439
↓ 8 callersMethodput
imperative/src/impl/persistent_cache.cpp:82
↓ 8 callersMethodrealpath
src/jit/impl/utils.cpp:210
↓ 8 callersMethodrecord
record an event on cn
imperative/src/impl/physical_tensor.cpp:320
↓ 8 callersMethodrend
dnn/include/megdnn/thin/small_vector.h:148
↓ 8 callersMethodreplace_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 callersFunctionrepr_pyobj
imperative/python/src/helper.cpp:66
↓ 8 callersFunctionrun_all_gather
src/opr/test/muxing.cpp:14
↓ 8 callersFunctionsafe_call
(func, qparams)
imperative/python/megengine/quantization/quantize.py:150
↓ 8 callersMethodset
dnn/src/rocm/convolution/helper.h:66
↓ 8 callersMethodset_log_level
imperative/python/src/utils.cpp:47
↓ 8 callersFunctionset_symbolic_shape
r"""Sets whether tensor.shape returns a tensor instead of a tuple
imperative/python/megengine/core/_trace_option.py:19
↓ 8 callersMethodshape
imperative/src/include/megbrain/imperative/basic_operators.h:107
↓ 8 callersMethodsize
src/core/include/megbrain/utils/thin/hash_table.h:357
↓ 8 callersMethodsize_log
size() in log2
dnn/include/megdnn/dtype.h:503
↓ 8 callersMethodstatic_device_memory_manager
set underlying device memory manager
src/core/impl/graph/var_node_mem_mgr.h:202
↓ 8 callersFunctionstbi__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 callersFunctionstbi__convert_format
lite/example/cpp_example/mge/cv/stb_image.h:1792
↓ 8 callersFunctionstbi__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 callersFunctionstbir__use_upsampling
lite/example/cpp_example/mge/cv/stb_image_resize.h:887
↓ 8 callersFunctionstbiw__putc
lite/example/cpp_example/mge/cv/stb_image_write.h:423
← previousnext →1,301–1,400 of 33,912, ranked by callers