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
↓ 14 callers
Method
assert_is
assert this dtype is given type; throw exception on failure
dnn/include/megdnn/dtype.h:510
↓ 14 callers
Method
cast
imperative/python/src/ops.cpp:501
↓ 14 callers
Method
current_scope
(self)
imperative/python/megengine/traced_module/module_tracer.py:150
↓ 14 callers
Method
desc
src/opr/impl/standalone/nms_cambricon.cpp:21
↓ 14 callers
Method
device
src/custom/impl/tensor.cpp:465
↓ 14 callers
Function
disable_observer
r"""Recursively disable ``module`` observer in QATModule through :meth:`~.Module.apply` Args: module: root module to do disable observer
imperative/python/megengine/quantization/quantize.py:281
↓ 14 callers
Method
encode
imperative/src/include/megbrain/imperative/graph_builder.h:91
↓ 14 callers
Function
evaluate_polynomial
src/opr/test/basic_arith/erfinv.h:55
↓ 14 callers
Function
evaluate_polynomial
dnn/src/common/elemwise/erfinv.h:55
↓ 14 callers
Method
execute
(self, *args, **kwargs)
imperative/python/megengine/jit/tracing.py:279
↓ 14 callers
Method
free_device
src/core/impl/comp_node/comp_node.cpp:449
↓ 14 callers
Function
func
(x, w, b, dy)
imperative/python/test/unit/xla/functional/test_xla_normalize.py:39
↓ 14 callers
Function
gconv_param
dnn/test/rocm/chanwise_convolution.cpp:28
↓ 14 callers
Function
get_bundle
dnn/src/arm_common/conv_bias/int8/direct_nchw44_algo.cpp:30
↓ 14 callers
Function
get_const_value
(expr: Expr, fall_back: Any = None)
imperative/python/megengine/traced_module/_passes/utils.py:20
↓ 14 callers
Method
get_default
lite/load_and_run/src/helpers/common.h:159
↓ 14 callers
Method
get_dev_tensor
imperative/src/impl/interpreter/interpreter_impl.cpp:551
↓ 14 callers
Method
get_mem_addr_alignment
get alignment requiement in bytes; guaranteed to be power of 2
src/core/include/megbrain/comp_node.h:285
↓ 14 callers
Method
get_output_name
get the output name by the index in the network Args: index: the index of the output name Returns:
lite/pylite/megenginelite/network.py:802
↓ 14 callers
Function
get_winograd_args
dnn/test/common/conv_bias.cpp:109
↓ 14 callers
Function
getter
src/opr/test/blas.cpp:232
↓ 14 callers
Function
init_dataset
()
imperative/python/test/unit/data/test_dataloader.py:40
↓ 14 callers
Method
instance_id
! * \brief get current hashed instance ID */
src/core/include/megbrain/graph/operator_node.h:88
↓ 14 callers
Method
is_available_attribute
dnn/src/cuda/conv_bias/algo.h:144
↓ 14 callers
Function
is_cudnn_supported
dnn/src/cuda/conv_bias/helper.cpp:126
↓ 14 callers
Method
is_non_overlapping_strong
dnn/src/common/basic_types.cpp:338
↓ 14 callers
Method
is_root
imperative/src/impl/interpreter/tensor_info.h:27
↓ 14 callers
Method
load
lite/load_and_run/src/helpers/json_loader.cpp:300
↓ 14 callers
Method
locator_logical
get the logical locator that created this comp node
src/core/include/megbrain/comp_node.h:448
↓ 14 callers
Function
log
r"""Element-wise :math:`\operatorname{log}(x)` function. Calculates an approximation to the natural (base :math:`e`) logarithm for each eleme
imperative/python/megengine/functional/elemwise.py:679
↓ 14 callers
Function
matmuls
src/opr/impl/blas.cpp:622
↓ 14 callers
Function
megcoreFree
dnn/src/common/megcore/public_api/device.cpp:71
↓ 14 callers
Function
megcoreMalloc
dnn/src/common/megcore/public_api/device.cpp:64
↓ 14 callers
Function
naive_transpose_kn_pad
dnn/src/x86/matrix_mul/common/common.h:413
↓ 14 callers
Function
name
dnn/src/cuda/batched_matrix_mul/algo.h:115
↓ 14 callers
Method
output
src/serialization/include/megbrain/serialization/opr_registry.h:19
↓ 14 callers
Function
record_event
imperative/src/impl/physical_tensor.cpp:495
↓ 14 callers
Function
ref
(x, y)
imperative/python/test/unit/functional/test_loss.py:29
↓ 14 callers
Method
relayout_opr
global relayout opr
dnn/src/cuda/handle.h:78
↓ 14 callers
Method
reshape
lite/src/tensor.cpp:124
↓ 14 callers
Function
reverse_adaptor
src/core/include/megbrain/utils/metahelper.h:190
↓ 14 callers
Function
sc
dnn/src/cuda/softmax/opr_impl.cpp:45
↓ 14 callers
Method
serialize
src/rdnn/impl/profiler.cpp:89
↓ 14 callers
Method
slice
lite/src/tensor.cpp:191
↓ 14 callers
Method
start
dnn/test/common/timer.h:20
↓ 14 callers
Method
sync
dnn/test/common/task_executor.cpp:192
↓ 14 callers
Method
to_c
(self)
imperative/python/megengine/core/_wrap.py:28
↓ 14 callers
Function
vaddlvq_s16
dnn/src/arm_common/simd_macro/marm_neon.h:348
↓ 14 callers
Method
vzero
dnn/src/arm_common/elemwise_helper/kimpl/fuse_add_relu.h:81
↓ 13 callers
Method
Concat
src/opr/impl/tensor_manip.cpp:1215
↓ 13 callers
Method
GetAPI
src/opr/impl/nvof/NvOFCuda.h:43
↓ 13 callers
Function
LITE_get_tensor_total_size_in_byte
lite/lite-c/src/tensor.cpp:185
↓ 13 callers
Function
LITE_make_network
lite/lite-c/src/network.cpp:209
↓ 13 callers
Function
MIDOUT_BEGIN
only enable midout for CPU, becuase CPU might be unused when some other platforms are used
dnn/src/common/handle.cpp:55
↓ 13 callers
Function
_dump_and_load
(func, symbolic, keep_opr_name=True)
imperative/python/test/unit/utils/test_dump_naming.py:17
↓ 13 callers
Function
_inplace
(f)
imperative/python/megengine/core/tensor/array_method.py:192
↓ 13 callers
Function
_unwrap
(x)
imperative/python/megengine/core/tensor/megbrain_graph.py:554
↓ 13 callers
Method
add_inputs
(self, vals)
imperative/python/megengine/traced_module/expr.py:118
↓ 13 callers
Method
add_stream
src/core/impl/comp_node/mem_alloc/impl.cpp:286
↓ 13 callers
Method
align_axis
dnn/include/megdnn/tensor_format.h:116
↓ 13 callers
Method
alignment_requirement
dnn/src/cambricon/handle.cpp:99
↓ 13 callers
Method
alloc_shared
! * \brief allocate as shared_ptr that binds the deallocator */
src/core/include/megbrain/comp_node/alloc.h:48
↓ 13 callers
Method
call_with_opr
src/gopt/impl/framework.cpp:426
↓ 13 callers
Method
capacity
dnn/include/megdnn/thin/small_vector.h:105
↓ 13 callers
Method
comp_node
imperative/src/include/megbrain/imperative/physical_tensor.h:129
↓ 13 callers
Function
deduce_layout
imperative/src/impl/ops/subtensor.cpp:85
↓ 13 callers
Function
default_network_io
lite/lite-c/src/network.cpp:62
↓ 13 callers
Method
destroy_range
dnn/include/megdnn/thin/small_vector.h:174
↓ 13 callers
Method
enabled
src/core/impl/graph/cg_impl.cpp:138
↓ 13 callers
Method
extra_info
! * \brief get associated extra info, or nullptr */
src/core/include/megbrain/exception.h:86
↓ 13 callers
Function
fmax
Maximum of half expressions. \param x first operand \param y second operand \return maximum of operands template<typename T,typename U> typename resul
dnn/include/megdnn/dtype/half.hpp:2208
↓ 13 callers
Function
fmin
Minimum of half expressions. \param x first operand \param y second operand \return minimum of operands template<typename T,typename U> typename resul
dnn/include/megdnn/dtype/half.hpp:2218
↓ 13 callers
Method
format
src/core/include/megbrain/graph/var_node.h:661
↓ 13 callers
Method
get_all_output_name
lite/src/network.cpp:254
↓ 13 callers
Function
get_bundle
dnn/src/arm_common/conv_bias/f16/direct_nchw88_algo.cpp:20
↓ 13 callers
Method
get_msecs
get milliseconds (one thousandth of a second)
src/core/include/megbrain/utils/timer.h:36
↓ 13 callers
Method
get_opr_type
dnn/include/megdnn/oprs/nn.h:938
↓ 13 callers
Method
inc_ref
imperative/src/include/megbrain/imperative/utils/local_ptr.h:21
↓ 13 callers
Method
input_waiting_spec
! * \brief set input waiting spec * * This should only be called from * SeqCompNodeOptimizerImpl::init_ready_event() or EagerEvalM
src/core/include/megbrain/graph/operator_node.h:571
↓ 13 callers
Function
interleave_8
dnn/src/aarch64/matrix_mul/asm/common.h:1306
↓ 13 callers
Function
invoke
src/core/impl/system.cpp:475
↓ 13 callers
Method
is_overwrite_root
! * \brief whether this interval does not override any other interval */
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/impl.h:164
↓ 13 callers
Function
l2_block_helper
block_helper is used to calculate oh block size
dnn/src/fallback/conv_bias/gi/block_helper.h:5
↓ 13 callers
Function
megcoreGetDeviceID
dnn/src/common/megcore/public_api/device.cpp:32
↓ 13 callers
Method
modules
r"""Returns an iterable for all the modules within this module, including itself.
imperative/python/megengine/module/module.py:349
↓ 13 callers
Method
nr_threads
dnn/src/common/megcore/cpu/default_computing_context.cpp:19
↓ 13 callers
Method
permutation
r"""Randomly permute a sequence, or return a permuted range. If ``n`` is a multi-dimensional tensor, it is only shuffled along its first i
imperative/python/megengine/random/rng.py:594
↓ 13 callers
Method
procedural_name
The full procedural name indicates architecture, extended name, tile size, and layout.
dnn/scripts/cutlass_generator/gemm_operation.py:179
↓ 13 callers
Method
register_as_builtin
(cls, mod)
imperative/python/megengine/traced_module/module_tracer.py:128
↓ 13 callers
Function
replace_vars
r"""Replaces vars in the graph. Args: dst: target vars representing the graph. varmap: the map that specifies how to replace the
imperative/python/megengine/utils/comp_graph_tools.py:280
↓ 13 callers
Method
round
dnn/src/fallback/quantized_converter.h:66
↓ 13 callers
Method
run
dnn/src/cuda/cutlass/gemm_operation.h:162
↓ 13 callers
Function
run_test
src/plugin/test/opr_footprint.cpp:45
↓ 13 callers
Function
runner
(is_trace)
imperative/python/test/unit/xla/test_xla_training.py:43
↓ 13 callers
Method
scale
src/custom/impl/tensor.cpp:332
↓ 13 callers
Method
share_memory_with
lite/src/tensor.cpp:209
↓ 13 callers
Method
sin
dnn/include/megdnn/dtype/half.hpp:1338
↓ 13 callers
Method
start
(self)
imperative/python/megengine/utils/profiler.py:177
↓ 13 callers
Method
stop
dnn/test/common/timer.h:26
↓ 13 callers
Function
sync
src/core/impl/comp_node_api.cpp:27
← previous
next →
901–1,000 of 33,912, ranked by callers