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
↓ 10 callers
Method
sample
(self, batch)
imperative/python/megengine/module/vision.py:567
↓ 10 callers
Method
set_device_memory_allocator
src/core/impl/graph/cg_impl.cpp:814
↓ 10 callers
Function
small_sort
src/core/include/megbrain/utils/metahelper.h:207
↓ 10 callers
Function
squeeze
r"""Removes dimension of shape 1. Args: inp: input tensor. axis: place of axis to be removed. Returns: output tensor
imperative/python/megengine/functional/tensor.py:1070
↓ 10 callers
Function
stbi__blinn_8x8
fast 0..255 * 0..255 => 0..255 rounded multiplication
lite/example/cpp_example/mge/cv/stb_image.h:4168
↓ 10 callers
Function
stbi__malloc_mad3
lite/example/cpp_example/mge/cv/stb_image.h:1081
↓ 10 callers
Function
stbir__resize_arbitrary
lite/example/cpp_example/mge/cv/stb_image_resize.h:2635
↓ 10 callers
Function
stream
src/custom/include/megbrain/custom/platform/custom_cuda.h:21
↓ 10 callers
Function
tester
(TestMod, ishape, dtype=None, atol=1e-5, **kwargs)
imperative/python/test/unit/xla/module/test_elemwise.py:19
↓ 10 callers
Method
to_bytes
src/custom/impl/param.cpp:139
↓ 10 callers
Method
wait
(self)
imperative/python/megengine/utils/custom_op_tools.py:613
↓ 9 callers
Function
GiStoreInt32
dnn/src/fallback/general_intrinsic/gi_int.h:214
↓ 9 callers
Function
LITE_reset_tensor_memory
lite/lite-c/src/tensor.cpp:90
↓ 9 callers
Function
XXH_readLE64
src/core/impl/utils/xxhash/xxhash.cpp:285
↓ 9 callers
Method
__init__
(self, left: ExprPattern, right: ExprPattern)
imperative/python/megengine/traced_module/_passes/pattern.py:121
↓ 9 callers
Method
__repr__
(self)
imperative/python/megengine/tensor.py:166
↓ 9 callers
Function
_register_supported_type
(type, flatten_fn=None, unflatten_fn=None)
imperative/python/megengine/traced_module/pytree.py:113
↓ 9 callers
Method
activate
src/core/impl/comp_node/atlas/comp_node.cpp:45
↓ 9 callers
Method
activate
src/core/impl/comp_node/cuda/comp_node.cpp:180
↓ 9 callers
Method
add_input
src/core/impl/graph/operator_node.cpp:216
↓ 9 callers
Method
alloc_output
dnn/test/common/utils.cpp:114
↓ 9 callers
Method
apply
src/core/impl/tensor.cpp:98
↓ 9 callers
Method
as_fwd_args
dnn/src/cuda/convolution/backward_data/algo.h:58
↓ 9 callers
Method
as_immutable_scalar_require_shape
src/core/impl/graph/symbol_var.cpp:77
↓ 9 callers
Function
as_tensor
(x)
imperative/python/test/unit/core/test_autodiff.py:38
↓ 9 callers
Method
broadcast
(self, shape: EinsumShape)
imperative/python/megengine/functional/einsum.py:51
↓ 9 callers
Function
build
(builder, outputs, outputs_has_grad)
imperative/python/megengine/core/tensor/utils.py:186
↓ 9 callers
Function
calculate_fan_in_and_fan_out
r"""Calculates fan_in / fan_out value for given weight tensor. This function assumes input tensor is stored in ``NCHW`` format. Note:
imperative/python/megengine/module/init.py:118
↓ 9 callers
Function
common_test_reduce
(opr, ref_opr)
imperative/python/test/unit/functional/test_math.py:14
↓ 9 callers
Method
compile
(self, *args)
imperative/python/megengine/core/tensor/megbrain_graph.py:55
↓ 9 callers
Function
concat
r"""Concat some tensors Args: inps: input tensors to concat. axis: over which dimension the tensors are concatenated. Default: 0
imperative/python/megengine/functional/tensor.py:564
↓ 9 callers
Function
copy_padding_kern
dnn/src/arm_common/conv_bias/f16/direct_nchw88_algo.cpp:36
↓ 9 callers
Function
copy_plane_in_bytes
dnn/src/common/utils.h:349
↓ 9 callers
Method
create_trt_network
src/tensorrt/test/make_trt_net.cpp:247
↓ 9 callers
Method
deduce_layout
dnn/src/naive/convpooling/opr_impl.cpp:56
↓ 9 callers
Method
device
imperative/python/src/tensor.cpp:688
↓ 9 callers
Method
end
src/core/include/megbrain/utils/metahelper.h:174
↓ 9 callers
Method
exec_fastrun
imperative/src/impl/dnn_op_helper.h:256
↓ 9 callers
Method
floor
dnn/include/megdnn/dtype/half.hpp:1574
↓ 9 callers
Method
free
src/core/test/mem_alloc.cpp:59
↓ 9 callers
Method
free
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/best_fit.cpp:57
↓ 9 callers
Function
from
dnn/src/naive/type_cvt/opr_impl.cpp:25
↓ 9 callers
Function
gen_dct_case
dnn/test/common/dct_ref.cpp:91
↓ 9 callers
Function
generate_data
()
imperative/python/test/unit/data/test_transform.py:15
↓ 9 callers
Method
get
dnn/src/naive/remap/opr_impl.cpp:39
↓ 9 callers
Function
get_client
r"""Get client of python XML RPC server.
imperative/python/megengine/distributed/group.py:234
↓ 9 callers
Function
get_expand_structure
r"""Gets Module's attribute compatible with complex key from Module's :meth:`~.named_children`. Supports handling structure containing list or dic
imperative/python/megengine/utils/module_utils.py:8
↓ 9 callers
Function
get_global_rng_seed
imperative/src/impl/ops/rng.cpp:1016
↓ 9 callers
Method
get_mem_status_bytes
get total and free memory on the computing device in bytes
src/core/include/megbrain/comp_node.h:387
↓ 9 callers
Function
get_nchw88_conv_bias_args
dnn/test/common/conv_bias.cpp:1673
↓ 9 callers
Method
get_raw
src/opr/include/megbrain/opr/internal/indexing_helper.h:26
↓ 9 callers
Method
get_uid
src/core/include/megbrain/comp_node.h:442
↓ 9 callers
Method
handle_type
dnn/include/megdnn/oprs/base.h:184
↓ 9 callers
Method
imag
imperative/src/include/megbrain/imperative/transformations/complex.h:44
↓ 9 callers
Function
infer_output_attrs_fallible
imperative/src/impl/ops/rng.cpp:875
↓ 9 callers
Method
infer_shape_fallible
src/core/impl/graph/static_infer_impl.cpp:845
↓ 9 callers
Function
interleave_8
dnn/src/armv7/matrix_mul/asm/common.h:646
↓ 9 callers
Function
interleave_helper
dnn/src/x86/matrix_mul/f32/strategy_6x16.cpp:488
↓ 9 callers
Method
is
imperative/src/include/megbrain/imperative/value.h:188
↓ 9 callers
Method
is_same
! * \brief Equality test */
src/core/include/megbrain/utils/hashable.h:28
↓ 9 callers
Method
lock
lock and acquire the operator
src/opr/include/megbrain/opr/utility.h:546
↓ 9 callers
Function
make
src/core/impl/graph/swap/swap_opr.h:44
↓ 9 callers
Function
make_divisible
src/gopt/test/network.cpp:334
↓ 9 callers
Function
make_no_value_infer
src/opr/include/megbrain/opr/io.h:182
↓ 9 callers
Function
make_tensor_descriptor
--------------- make cudnn-frontend tensor descriptor --------------- */
dnn/src/cuda/cudnn_wrapper_v8.cpp:112
↓ 9 callers
Function
maximum
r"""Element-wise maximum of tensor elements. Compare two tensors and returns a new tensor containing the element-wise maxima. If one of t
imperative/python/megengine/functional/elemwise.py:965
↓ 9 callers
Function
megcoreDestroyComputingHandle
dnn/src/common/megcore/public_api/computing.cpp:19
↓ 9 callers
Method
meta
(self)
imperative/python/megengine/data/dataset/vision/mnist.py:123
↓ 9 callers
Method
named_modules
r"""Returns an iterable of key-module pairs for all the modules within this module, including itself, where 'key' is the dotted path from this
imperative/python/megengine/module/module.py:357
↓ 9 callers
Method
nr_alive
src/core/test/graph/partial_execution.cpp:80
↓ 9 callers
Method
offset
dnn/test/common/index.h:24
↓ 9 callers
Method
op_name_list
src/custom/impl/manager.cpp:161
↓ 9 callers
Function
output_info
lite/example/cpp_example/mge/basic.cpp:12
↓ 9 callers
Function
propagate_qconfig
r"""Recursively set ``module``'s qconfig through :meth:`~.Module.apply`. Args: module: root module to traverse recursively. qconf
imperative/python/megengine/quantization/quantize.py:180
↓ 9 callers
Method
push
src/core/impl/utils/thread.cpp:262
↓ 9 callers
Method
read
src/serialization/impl/file.cpp:41
↓ 9 callers
Function
rep
dnn/src/fallback/split/opr_impl.cpp:15
↓ 9 callers
Function
rep
dnn/src/naive/local/opr_impl.cpp:37
↓ 9 callers
Function
rep
dnn/src/naive/pooling/opr_impl.cpp:280
↓ 9 callers
Method
reset
lite/load_and_run/src/helpers/text_table.h:90
↓ 9 callers
Function
run
( N, IC, OC, IH, IW, KH, KW, PH, PW,
imperative/python/test/unit/functional/test_functional.py:793
↓ 9 callers
Function
save
r"""Save an object to disk file. The saved object must be a :class:`~.module.Module`, :attr:`.Module.state_dict` or :attr:`.Optimizer.state_di
imperative/python/megengine/serialization.py:9
↓ 9 callers
Method
set
(self, value)
imperative/python/megengine/utils/future.py:9
↓ 9 callers
Method
set_device_id
lite/src/mge/network_impl.cpp:203
↓ 9 callers
Method
sleep
src/opr/impl/utility.cpp:59
↓ 9 callers
Method
solve
src/gopt/impl/global_layout_transform/profiling_based_solver.cpp:54
↓ 9 callers
Function
stbir__edge_wrap
lite/example/cpp_example/mge/cv/stb_image_resize.h:1010
↓ 9 callers
Function
stbiw__jpg_processDU
lite/example/cpp_example/mge/cv/stb_image_write.h:1507
↓ 9 callers
Function
stbiw__jpg_writeBits
lite/example/cpp_example/mge/cv/stb_image_write.h:1425
↓ 9 callers
Method
to_numpy
Convert the result tensor to a numpy ndarray
lite/pylite/megenginelite/utils.py:227
↓ 9 callers
Function
transpose_4
dnn/src/aarch64/matrix_mul/asm/common.h:1983
↓ 9 callers
Function
transpose_8
dnn/src/aarch64/matrix_mul/asm/common.h:2003
↓ 9 callers
Method
unwrap_input
imperative/src/impl/transformations/format.cpp:69
↓ 9 callers
Method
update
lite/load_and_run/src/options/optimize_options.cpp:38
↓ 9 callers
Function
usable
dnn/src/x86/conv_bias/f32/algos.h:120
↓ 9 callers
Method
use_count
shortcut for raw_storage().use_count(), but won't trigger lazy alloc
src/core/include/megbrain/tensor.h:252
↓ 9 callers
Function
vmlal_high_s8
dnn/src/arm_common/simd_macro/marm_neon.h:357
↓ 9 callers
Method
wait
imperative/src/impl/transformations/trace.cpp:859
↓ 9 callers
Method
wait
wait until forward finish in sync model
lite/pylite/megenginelite/network.py:635
↓ 8 callers
Function
MIDOUT_BEGIN
dnn/src/common/warp_common.h:434
← previous
next →
1,201–1,300 of 33,912, ranked by callers