MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 10 callersMethodsample
(self, batch)
imperative/python/megengine/module/vision.py:567
↓ 10 callersMethodset_device_memory_allocator
src/core/impl/graph/cg_impl.cpp:814
↓ 10 callersFunctionsmall_sort
src/core/include/megbrain/utils/metahelper.h:207
↓ 10 callersFunctionsqueeze
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 callersFunctionstbi__blinn_8x8
fast 0..255 * 0..255 => 0..255 rounded multiplication
lite/example/cpp_example/mge/cv/stb_image.h:4168
↓ 10 callersFunctionstbi__malloc_mad3
lite/example/cpp_example/mge/cv/stb_image.h:1081
↓ 10 callersFunctionstbir__resize_arbitrary
lite/example/cpp_example/mge/cv/stb_image_resize.h:2635
↓ 10 callersFunctionstream
src/custom/include/megbrain/custom/platform/custom_cuda.h:21
↓ 10 callersFunctiontester
(TestMod, ishape, dtype=None, atol=1e-5, **kwargs)
imperative/python/test/unit/xla/module/test_elemwise.py:19
↓ 10 callersMethodto_bytes
src/custom/impl/param.cpp:139
↓ 10 callersMethodwait
(self)
imperative/python/megengine/utils/custom_op_tools.py:613
↓ 9 callersFunctionGiStoreInt32
dnn/src/fallback/general_intrinsic/gi_int.h:214
↓ 9 callersFunctionLITE_reset_tensor_memory
lite/lite-c/src/tensor.cpp:90
↓ 9 callersFunctionXXH_readLE64
src/core/impl/utils/xxhash/xxhash.cpp:285
↓ 9 callersMethod__init__
(self, left: ExprPattern, right: ExprPattern)
imperative/python/megengine/traced_module/_passes/pattern.py:121
↓ 9 callersMethod__repr__
(self)
imperative/python/megengine/tensor.py:166
↓ 9 callersFunction_register_supported_type
(type, flatten_fn=None, unflatten_fn=None)
imperative/python/megengine/traced_module/pytree.py:113
↓ 9 callersMethodactivate
src/core/impl/comp_node/atlas/comp_node.cpp:45
↓ 9 callersMethodactivate
src/core/impl/comp_node/cuda/comp_node.cpp:180
↓ 9 callersMethodadd_input
src/core/impl/graph/operator_node.cpp:216
↓ 9 callersMethodalloc_output
dnn/test/common/utils.cpp:114
↓ 9 callersMethodapply
src/core/impl/tensor.cpp:98
↓ 9 callersMethodas_fwd_args
dnn/src/cuda/convolution/backward_data/algo.h:58
↓ 9 callersMethodas_immutable_scalar_require_shape
src/core/impl/graph/symbol_var.cpp:77
↓ 9 callersFunctionas_tensor
(x)
imperative/python/test/unit/core/test_autodiff.py:38
↓ 9 callersMethodbroadcast
(self, shape: EinsumShape)
imperative/python/megengine/functional/einsum.py:51
↓ 9 callersFunctionbuild
(builder, outputs, outputs_has_grad)
imperative/python/megengine/core/tensor/utils.py:186
↓ 9 callersFunctioncalculate_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 callersFunctioncommon_test_reduce
(opr, ref_opr)
imperative/python/test/unit/functional/test_math.py:14
↓ 9 callersMethodcompile
(self, *args)
imperative/python/megengine/core/tensor/megbrain_graph.py:55
↓ 9 callersFunctionconcat
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 callersFunctioncopy_padding_kern
dnn/src/arm_common/conv_bias/f16/direct_nchw88_algo.cpp:36
↓ 9 callersFunctioncopy_plane_in_bytes
dnn/src/common/utils.h:349
↓ 9 callersMethodcreate_trt_network
src/tensorrt/test/make_trt_net.cpp:247
↓ 9 callersMethoddeduce_layout
dnn/src/naive/convpooling/opr_impl.cpp:56
↓ 9 callersMethoddevice
imperative/python/src/tensor.cpp:688
↓ 9 callersMethodend
src/core/include/megbrain/utils/metahelper.h:174
↓ 9 callersMethodexec_fastrun
imperative/src/impl/dnn_op_helper.h:256
↓ 9 callersMethodfloor
dnn/include/megdnn/dtype/half.hpp:1574
↓ 9 callersMethodfree
src/core/test/mem_alloc.cpp:59
↓ 9 callersMethodfree
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/best_fit.cpp:57
↓ 9 callersFunctionfrom
dnn/src/naive/type_cvt/opr_impl.cpp:25
↓ 9 callersFunctiongen_dct_case
dnn/test/common/dct_ref.cpp:91
↓ 9 callersFunctiongenerate_data
()
imperative/python/test/unit/data/test_transform.py:15
↓ 9 callersMethodget
dnn/src/naive/remap/opr_impl.cpp:39
↓ 9 callersFunctionget_client
r"""Get client of python XML RPC server.
imperative/python/megengine/distributed/group.py:234
↓ 9 callersFunctionget_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 callersFunctionget_global_rng_seed
imperative/src/impl/ops/rng.cpp:1016
↓ 9 callersMethodget_mem_status_bytes
get total and free memory on the computing device in bytes
src/core/include/megbrain/comp_node.h:387
↓ 9 callersFunctionget_nchw88_conv_bias_args
dnn/test/common/conv_bias.cpp:1673
↓ 9 callersMethodget_raw
src/opr/include/megbrain/opr/internal/indexing_helper.h:26
↓ 9 callersMethodget_uid
src/core/include/megbrain/comp_node.h:442
↓ 9 callersMethodhandle_type
dnn/include/megdnn/oprs/base.h:184
↓ 9 callersMethodimag
imperative/src/include/megbrain/imperative/transformations/complex.h:44
↓ 9 callersFunctioninfer_output_attrs_fallible
imperative/src/impl/ops/rng.cpp:875
↓ 9 callersMethodinfer_shape_fallible
src/core/impl/graph/static_infer_impl.cpp:845
↓ 9 callersFunctioninterleave_8
dnn/src/armv7/matrix_mul/asm/common.h:646
↓ 9 callersFunctioninterleave_helper
dnn/src/x86/matrix_mul/f32/strategy_6x16.cpp:488
↓ 9 callersMethodis
imperative/src/include/megbrain/imperative/value.h:188
↓ 9 callersMethodis_same
! * \brief Equality test */
src/core/include/megbrain/utils/hashable.h:28
↓ 9 callersMethodlock
lock and acquire the operator
src/opr/include/megbrain/opr/utility.h:546
↓ 9 callersFunctionmake
src/core/impl/graph/swap/swap_opr.h:44
↓ 9 callersFunctionmake_divisible
src/gopt/test/network.cpp:334
↓ 9 callersFunctionmake_no_value_infer
src/opr/include/megbrain/opr/io.h:182
↓ 9 callersFunctionmake_tensor_descriptor
--------------- make cudnn-frontend tensor descriptor --------------- */
dnn/src/cuda/cudnn_wrapper_v8.cpp:112
↓ 9 callersFunctionmaximum
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 callersFunctionmegcoreDestroyComputingHandle
dnn/src/common/megcore/public_api/computing.cpp:19
↓ 9 callersMethodmeta
(self)
imperative/python/megengine/data/dataset/vision/mnist.py:123
↓ 9 callersMethodnamed_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 callersMethodnr_alive
src/core/test/graph/partial_execution.cpp:80
↓ 9 callersMethodoffset
dnn/test/common/index.h:24
↓ 9 callersMethodop_name_list
src/custom/impl/manager.cpp:161
↓ 9 callersFunctionoutput_info
lite/example/cpp_example/mge/basic.cpp:12
↓ 9 callersFunctionpropagate_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 callersMethodpush
src/core/impl/utils/thread.cpp:262
↓ 9 callersMethodread
src/serialization/impl/file.cpp:41
↓ 9 callersFunctionrep
dnn/src/fallback/split/opr_impl.cpp:15
↓ 9 callersFunctionrep
dnn/src/naive/local/opr_impl.cpp:37
↓ 9 callersFunctionrep
dnn/src/naive/pooling/opr_impl.cpp:280
↓ 9 callersMethodreset
lite/load_and_run/src/helpers/text_table.h:90
↓ 9 callersFunctionrun
( N, IC, OC, IH, IW, KH, KW, PH, PW,
imperative/python/test/unit/functional/test_functional.py:793
↓ 9 callersFunctionsave
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 callersMethodset
(self, value)
imperative/python/megengine/utils/future.py:9
↓ 9 callersMethodset_device_id
lite/src/mge/network_impl.cpp:203
↓ 9 callersMethodsleep
src/opr/impl/utility.cpp:59
↓ 9 callersMethodsolve
src/gopt/impl/global_layout_transform/profiling_based_solver.cpp:54
↓ 9 callersFunctionstbir__edge_wrap
lite/example/cpp_example/mge/cv/stb_image_resize.h:1010
↓ 9 callersFunctionstbiw__jpg_processDU
lite/example/cpp_example/mge/cv/stb_image_write.h:1507
↓ 9 callersFunctionstbiw__jpg_writeBits
lite/example/cpp_example/mge/cv/stb_image_write.h:1425
↓ 9 callersMethodto_numpy
Convert the result tensor to a numpy ndarray
lite/pylite/megenginelite/utils.py:227
↓ 9 callersFunctiontranspose_4
dnn/src/aarch64/matrix_mul/asm/common.h:1983
↓ 9 callersFunctiontranspose_8
dnn/src/aarch64/matrix_mul/asm/common.h:2003
↓ 9 callersMethodunwrap_input
imperative/src/impl/transformations/format.cpp:69
↓ 9 callersMethodupdate
lite/load_and_run/src/options/optimize_options.cpp:38
↓ 9 callersFunctionusable
dnn/src/x86/conv_bias/f32/algos.h:120
↓ 9 callersMethoduse_count
shortcut for raw_storage().use_count(), but won't trigger lazy alloc
src/core/include/megbrain/tensor.h:252
↓ 9 callersFunctionvmlal_high_s8
dnn/src/arm_common/simd_macro/marm_neon.h:357
↓ 9 callersMethodwait
imperative/src/impl/transformations/trace.cpp:859
↓ 9 callersMethodwait
wait until forward finish in sync model
lite/pylite/megenginelite/network.py:635
↓ 8 callersFunctionMIDOUT_BEGIN
dnn/src/common/warp_common.h:434
← previousnext →1,201–1,300 of 33,912, ranked by callers