MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 17 callersMethoddispatch
src/core/impl/comp_node/cpu/comp_node.cpp:428
↓ 17 callersMethoddo_forward
(self, network, times=3)
lite/pylite/test/test_network_device.py:43
↓ 17 callersMethodget
src/core/impl/comp_node/comp_node.cpp:339
↓ 17 callersFunctionget_aligned_power2
src/core/include/megbrain/utils/arith_helper.h:46
↓ 17 callersFunctionget_mean_var
dnn/test/naive/rng.h:18
↓ 17 callersMethodget_time_in_us
dnn/test/common/timer.h:32
↓ 17 callersMethodhandle
src/core/include/megbrain/comp_node_env.h:629
↓ 17 callersMethodinit_from_given_tensor
dnn/src/common/elemwise_helper.cpp:7
↓ 17 callersMethodinit_output_static_infer_desc_workspace
src/opr/include/megbrain/opr/internal/megdnn_opr_wrapper.h:193
↓ 17 callersMethodinsert_exprs
r"""Initialize the trace mode and insertion position. When used within a 'with' statement, this will temporary set the trace mode and
imperative/python/megengine/traced_module/traced_module.py:984
↓ 17 callersMethodinst
imperative/src/impl/ops/custom_opdef.cpp:125
↓ 17 callersMethodmegcore_dispatcher
dnn/src/naive/handle.h:169
↓ 17 callersMethodmut_desc
Use CnnlDescriptor() to access the underlying desciptor pointer if you intend to modify what it points to This will ensure that the descriptor is init
dnn/src/cambricon/cnnl_wrapper/cnnl_common_descriptors.h:39
↓ 17 callersFunctionquantize_qat
r"""Recursively convert float :class:`~.Module` to :class:`~.QATModule` through :meth:`~.Module.apply` and set qconfig relatively. Args:
imperative/python/megengine/quantization/quantize.py:84
↓ 17 callersMethodregiste_fixed_options
lite/load_and_run/src/strategys/strategy_fitting.cpp:120
↓ 17 callersFunctionround
r"""Element-wise :math:`\operatorname{round}(x)` function. Rounds each element :math:`x_i` of the input tensor :math:`x` to the nearest integ
imperative/python/megengine/functional/elemwise.py:852
↓ 17 callersMethodshared_from_this
imperative/src/include/megbrain/imperative/op_def.h:94
↓ 17 callersFunctiontest
(func, inp, comp, target_shp)
imperative/python/test/unit/functional/test_tensor.py:421
↓ 16 callersFunctionMEGDNN_FOREACH_PARAMETERIZED_DTYPE
src/gopt/impl/global_layout_transform/reformat_manager.cpp:77
↓ 16 callersFunctionMGB_CATCH
src/core/impl/system.cpp:594
↓ 16 callersMethod__init__
(self)
imperative/python/test/unit/module/test_module.py:33
↓ 16 callersFunction_subtensor
(src, axis, begin, end)
imperative/python/megengine/module/vision.py:274
↓ 16 callersFunctionabs
r"""Element-wise :math:`\operatorname{abs}(x)` function. Calculates the absolute value for each element :math:`x_i` of the input tensor :math
imperative/python/megengine/functional/elemwise.py:557
↓ 16 callersMethodapply_quant_activation
r"""Apply weight's observer and fake_quant from ``qconfig`` on ``target``.
imperative/python/megengine/module/qat/module.py:95
↓ 16 callersMethodbackward
imperative/python/src/grad.cpp:50
↓ 16 callersFunctionbenchmark_conv1x1
dnn/test/arm_common/conv_bias.cpp:2250
↓ 16 callersFunctionbenchmark_winograd
dnn/test/x86/conv_bias.cpp:3067
↓ 16 callersMethodcheck_attribute
dnn/src/common/algo_base.cpp:47
↓ 16 callersMethodcomp_node
imperative/src/impl/proxy_graph/mini_graph.h:64
↓ 16 callersMethoddeduce_dtype
dnn/src/common/conv_bias.cpp:137
↓ 16 callersFunctiondivup
src/core/include/megbrain/utils/arith_helper.h:13
↓ 16 callersMethoddnn_tensor
imperative/src/impl/physical_tensor.cpp:705
↓ 16 callersMethoddump
imperative/src/impl/interpreter/stack_manager.h:157
↓ 16 callersFunctionfma3
src/gopt/test/basic_arith.cpp:44
↓ 16 callersFunctionfunc
(x, w, b, dy)
imperative/python/test/unit/xla/functional/test_xla_nn.py:37
↓ 16 callersFunctionfunc
(lhs, rhs, dout)
imperative/python/test/unit/xla/functional/test_xla_math.py:38
↓ 16 callersFunctiongconv_param
dnn/test/cuda/chanwise_convolution3d.cpp:26
↓ 16 callersFunctionget_all_algorithms
dnn/src/common/algo_chooser.h:72
↓ 16 callersMethodget_backend_type
lite/src/mge/tensor_impl.h:39
↓ 16 callersFunctionget_nearest_linear_coord
dnn/src/arm_common/resize/helper.h:89
↓ 16 callersFunctionget_workspace_bundle
dnn/src/arm_common/lstm/lstm_utils.h:205
↓ 16 callersFunctionhandle_padding
dnn/src/arm_common/pooling/algo.cpp:63
↓ 16 callersFunctionif
src/rdnn/impl/algo_chooser.cpp:1190
↓ 16 callersMethodinfer_type
dnn/src/common/non_zero.cpp:15
↓ 16 callersMethodis
imperative/src/impl/value.cpp:36
↓ 16 callersFunctionis_distributed
r"""Return True if the distributed process group has been initialized.
imperative/python/megengine/distributed/group.py:196
↓ 16 callersMethodis_lowbit_aligned
dnn/src/common/tensor_format.cpp:74
↓ 16 callersMethodlog
dnn/include/megdnn/dtype/half.hpp:1232
↓ 16 callersFunctionmake_func_pair
make a pair of functions with and without JIT optimization
src/jit/test/fusion.cpp:86
↓ 16 callersMethodmegdnn_opr
src/opr/include/megbrain/opr/internal/megdnn_opr_wrapper.h:85
↓ 16 callersMethodname
get the name of IO item
lite/pylite/megenginelite/network.py:333
↓ 16 callersFunctionones
(shape, dtype=np.float32)
imperative/python/megengine/xla/rules/tensor.py:199
↓ 16 callersMethodoutput_captured
imperative/src/include/megbrain/imperative/transformations/grad.h:69
↓ 16 callersMethodrbegin
reverse iterator creation method.
dnn/include/megdnn/thin/small_vector.h:146
↓ 16 callersMethodread
(self, name)
imperative/python/megengine/xla/lib/config.py:80
↓ 16 callersMethodreset
dnn/src/common/basic_types.cpp:597
↓ 16 callersFunctionrun
( N, IC, OC, IH, IW, KH, KW, PH, PW,
imperative/python/test/unit/quantization/test_op.py:75
↓ 16 callersMethodstream
dnn/src/cuda/relayout/opr_impl.h:37
↓ 16 callersMethodsync
src/plugin/impl/opr_io_dump.cpp:140
↓ 16 callersMethodto_string_logical
get string representation of logical device
src/core/include/megbrain/comp_node.h:438
↓ 16 callersMethodtype
dnn/src/cambricon/conv_bias/algo.h:87
↓ 16 callersFunctionwrite_merge_file_name
(f, gen_op, gen_type, split_number)
dnn/scripts/cutlass_generator/gen_list.py:40
↓ 15 callersMethodadd_opr
src/jit/impl/internal_graph.cpp:176
↓ 15 callersMethodalloc
src/core/test/mem_alloc.cpp:44
↓ 15 callersMethodattribute
src/gopt/include/megbrain/gopt/layout_transform_context.h:167
↓ 15 callersFunctionbit
dnn/src/x86/utils.cpp:39
↓ 15 callersMethodconcat
lite/src/tensor.cpp:291
↓ 15 callersFunctiondefault_config
lite/lite-c/src/network.cpp:48
↓ 15 callersMethoddispatch
src/core/include/megbrain/comp_node_env.h:538
↓ 15 callersMethoddshape
dnn/test/common/local.h:27
↓ 15 callersMethoddump_buf_with_len
src/serialization/impl/serializer_oss.cpp:467
↓ 15 callersMethodexec
dnn/test/common/exec_proxy.h:75
↓ 15 callersFunctionfind
dnn/include/megdnn/thin/small_vector.h:851
↓ 15 callersMethodfshape
dnn/test/common/local.h:26
↓ 15 callersMethodgenerate
src/jit/impl/internal_graph.cpp:157
↓ 15 callersMethodgetEnumName
imperative/tablegen/helper.h:71
↓ 15 callersMethodgrow
dnn/include/megdnn/thin/small_vector.h:220
↓ 15 callersMethodid_str
src/core/impl/graph/var_node.cpp:173
↓ 15 callersFunctioninvoke_op
(op, inputs_, cvt_inputs=canonize_inputs)
imperative/python/test/unit/core/test_indexing_op.py:129
↓ 15 callersFunctioniota
do some thing like arange. for example: shape = (2, 3), dimension=1, output is [[0, 1, 2], [0, 1, 2]] shape = (2, 3), dimensi
imperative/python/megengine/xla/rules/tensor.py:231
↓ 15 callersMethoditem
imperative/src/include/megbrain/imperative/basic_values.h:103
↓ 15 callersMethodkey
(self)
imperative/python/megengine/distributed/group.py:68
↓ 15 callersFunctionload_library
dnn/cuda-stub/src/libnvrtc-wrap.h:238
↓ 15 callersFunctionmake_bundle
dnn/src/cambricon/elemwise/opr_impl.cpp:668
↓ 15 callersMethodmatch
(self, pattern: ExprPattern, expr: Expr)
imperative/python/megengine/traced_module/_passes/matcher.py:39
↓ 15 callersFunctionmodel
src/opr/include/megbrain/opr/atlas_runtime_op.h:40
↓ 15 callersFunctionname
dnn/src/x86/matrix_mul/algos.h:11
↓ 15 callersMethodnumpy
imperative/src/impl/value.cpp:44
↓ 15 callersMethodopen
imperative/src/impl/persistent_cache.cpp:135
↓ 15 callersMethodptr
dnn/src/common/cv/common.h:156
↓ 15 callersMethodsshape
dnn/test/common/local.h:25
↓ 15 callersFunctionstack
(inps, axis)
imperative/python/megengine/xla/rules/tensor.py:126
↓ 15 callersFunctionstbi__at_eof
lite/example/cpp_example/mge/cv/stb_image.h:1662
↓ 15 callersMethodtell
src/serialization/impl/file.cpp:46
↓ 15 callersMethodwhat
src/core/include/megbrain/exception.h:81
↓ 14 callersFunctionGiLoadInt8
dnn/src/fallback/general_intrinsic/gi_int.h:71
↓ 14 callersFunctionMIDOUT_BEGIN
dnn/src/naive/relayout_format/opr_impl.cpp:667
↓ 14 callersFunction_mm256_storeu2_m128_emulate
dnn/src/x86/avx_helper.h:27
↓ 14 callersMethodadd
src/core/test/mem_alloc.cpp:106
↓ 14 callersMethodalloc_device
src/core/impl/comp_node/comp_node.cpp:443
← previousnext →801–900 of 33,912, ranked by callers