MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 46 callersFunctionadd
r"""Element-wise addition. Calculates the sum for each element :math:`x_i` of the input tensor :math:`x` with the respective element :math:`y
imperative/python/megengine/functional/elemwise.py:76
↓ 46 callersMethodfind
dnn/include/megdnn/thin/small_vector.h:160
↓ 46 callersMethodinfer_value
imperative/src/impl/proxy_graph/mini_graph.h:520
↓ 46 callersMethodmegcore_computing_handle
! * \brief Get the underlying megcore computing handle. */
dnn/include/megdnn/handle.h:107
↓ 46 callersMethodname
dnn/src/naive/matrix_mul/algorithms.h:9
↓ 45 callersFunction_mm256_loadu2_m128_emulate
dnn/src/x86/avx_helper.h:20
↓ 45 callersFunctionpowf
src/opr/include/megbrain/opr/basic_arith_wrapper.h:71
↓ 45 callersMethodread
dnn/src/common/cv/mat.cpp:129
↓ 45 callersFunctiontester
(x_shape, y_shape, indices, dtype=None)
imperative/python/test/unit/xla/functional/test_xla_indexing.py:104
↓ 45 callersMethodtype
src/custom/impl/param.cpp:71
↓ 44 callersMethodflatten
r"""Get a new TracedModule, which eliminates ``GetAttr`` and has no hierarchy. Retruns: A new :class:`TracedModule`.
imperative/python/megengine/traced_module/traced_module.py:2159
↓ 44 callersFunctionget_workspace
dnn/src/x86/matrix_mul/algos.h:13
↓ 44 callersFunctionget_workspace_in_bytes
dnn/src/naive/rnn/funcs.h:23
↓ 44 callersFunctionis_static_var_shape
! * \brief whether var shape could be statically inferred */
src/core/include/megbrain/graph/helper.h:22
↓ 44 callersMethodmax
(self, axis, keepdims=False)
imperative/python/megengine/xla/rules/hlotensor.py:186
↓ 44 callersMethodnumpy
imperative/python/src/tensor.h:73
↓ 44 callersFunctionopr_test
:param cases: the list which have dict element, the list length should be 2 for dynamic shape test. and the dict should have input,
imperative/python/test/helpers/utils.py:40
↓ 44 callersMethodset_value
lite/load_and_run/src/helpers/common.h:157
↓ 44 callersFunctionstbi__rewind
lite/example/cpp_example/mge/cv/stb_image.h:900
↓ 44 callersMethodtranspose
(self, permutation)
imperative/python/megengine/xla/rules/hlotensor.py:151
↓ 43 callersMethodclear_grad
(self)
imperative/python/test/unit/jit/test_tracing.py:873
↓ 43 callersMethoddtype
(self)
imperative/python/megengine/xla/rules/hlotensor.py:52
↓ 43 callersFunctionlog
Natural logorithm. \param arg function argument \return logarithm of \a arg to base e template<typename T> typename enable<expr,T>::type log(T arg) {
dnn/include/megdnn/dtype/half.hpp:2267
↓ 43 callersMethodmake_rewriter
src/gopt/include/megbrain/gopt/framework.h:129
↓ 42 callersFunctionGiSt1Float32
dnn/src/fallback/general_intrinsic/gi_float.h:366
↓ 42 callersMethodexec
dnn/src/naive/rnn/opr_impl.cpp:23
↓ 42 callersMethodget_secs
src/core/include/megbrain/utils/timer.h:33
↓ 42 callersMethodmake_scalar_dt
src/core/include/megbrain/graph/symbol_var.h:112
↓ 42 callersMethodmul
dnn/src/common/winograd/winograd_generator.cpp:179
↓ 42 callersMethodreplace
(self, **kw)
imperative/python/megengine/xla/lower.py:70
↓ 42 callersMethodstep
(self)
imperative/python/megengine/utils/profiler.py:198
↓ 42 callersFunctionzeros
(shape, dtype=np.float32)
imperative/python/megengine/xla/rules/tensor.py:203
↓ 41 callersMethodall
(self, axis, keepdims=False)
imperative/python/megengine/xla/rules/hlotensor.py:196
↓ 41 callersMethoddist_elem
dnn/include/megdnn/basic_types.h:152
↓ 41 callersMethodinfo
imperative/src/impl/profiler/states.h:38
↓ 41 callersFunctionmake_call
src/jit/include/megbrain/jit/ast_c.h:189
↓ 41 callersMethodoutput_dtype
src/core/include/megbrain/graph/operator_node.h:115
↓ 41 callersFunctionsum
r"""Calculates the sum of tensor elements over a given axis (or axes). Args: inp: input tensor. Should have a numeric data type.
imperative/python/megengine/functional/math.py:150
↓ 41 callersMethodsymshape
src/core/impl/graph/symbol_var.cpp:17
↓ 41 callersMethodto_numpy
get the buffer of the tensor
lite/pylite/megenginelite/tensor.py:515
↓ 41 callersMethodtotal_nr_elems
imperative/src/include/megbrain/imperative/utils/value_shape.h:38
↓ 41 callersFunctiontry_cast
lite/include/lite/tensor.h:440
↓ 40 callersMethodcollapse_contiguous
* \brief The collapse_contiguous function will convert a contiguous image like * tensor layout into a 2-dimensional layout, shape[0] = height of the
dnn/src/common/basic_types.cpp:334
↓ 40 callersFunctionget_ABC
dnn/src/common/reduce_helper.cpp:10
↓ 40 callersFunctionget_device_count
r"""Gets number of devices installed on this system. Args: device_type: device type, one of 'gpu' or 'cpu'
imperative/python/megengine/device.py:89
↓ 40 callersMethodget_rank
src/opr-mm/include/megbrain/opr/group_manager.h:34
↓ 40 callersFunctionget_workspace_in_bytes
dnn/src/cuda/batched_matrix_mul/algo.h:108
↓ 40 callersMethodname
src/tensorrt/impl/opr_replace.cpp:1602
↓ 40 callersFunctionprefetch_3x
dnn/src/aarch64/matrix_mul/asm/common.h:57
↓ 40 callersMethodseed
(self)
imperative/python/megengine/module/vision.py:102
↓ 40 callersMethodstore
dnn/src/arm_common/resize/helper.h:21
↓ 40 callersFunctiontrace_module
r"""Traces module ``mod`` and returns corresponding :class:`TracedModule`. Args: mod(:attr:`.module.Module`): the module will be converte
imperative/python/megengine/traced_module/traced_module.py:2442
↓ 39 callersFunctioncopy
r"""Copies tensor to another device. Args: inp: input tensor. device: destination device. Examples: >>> import nump
imperative/python/megengine/functional/tensor.py:1209
↓ 39 callersFunctioncudnn_handle
dnn/src/cuda/utils.h:21
↓ 39 callersFunctionmgb_lar
lite/test/test_common.h:117
↓ 39 callersFunctionnext
! * \brief get the interval whose address is greater than newly * allocated one, or nullptr if the new interval is at end */
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/pushdown.cpp:170
↓ 39 callersMethodproperty
src/jit/impl/mlir/compiler.h:23
↓ 39 callersFunctionstbi__get16be
lite/example/cpp_example/mge/cv/stb_image.h:1732
↓ 39 callersMethodtrain
r"""Sets training mode of all the modules within this module (including itself) to ``mode``. This effectively sets the ``training`` attributes
imperative/python/megengine/module/module.py:392
↓ 39 callersMethodvalid
src/gopt/impl/global_layout_transform/dynamic_programming_solver.cpp:30
↓ 38 callersMethodid
imperative/src/impl/interpreter/stack_manager.h:136
↓ 38 callersFunctionmake
src/core/test/sublinear_memory.cpp:71
↓ 38 callersMethodname
src/serialization/impl/serializer_oss_common.h:44
↓ 38 callersMethodreset
src/opr/impl/loop/impl.h:417
↓ 38 callersMethodto_string
src/core/impl/graph/cg_impl.cpp:879
↓ 37 callersMethodcopy
imperative/python/src/tensor.h:51
↓ 37 callersFunctionf
(*args)
imperative/python/test/unit/core/test_autodiff.py:25
↓ 37 callersMethodget
imperative/src/impl/persistent_cache.cpp:60
↓ 37 callersMethodget
src/opr/test/dnn/batch_norm.cpp:34
↓ 37 callersFunctionis_op
(target)
imperative/python/megengine/traced_module/_passes/pattern.py:221
↓ 37 callersFunctionmegcoreSynchronize
dnn/src/common/megcore/public_api/computing.cpp:54
↓ 36 callersMethoddtype
(self)
lite/pylite/megenginelite/tensor.py:107
↓ 36 callersMethodencode
lite/load_and_run/src/helpers/utils.cpp:56
↓ 36 callersMethodexec
dnn/test/rocm/chanwise_convolution.cpp:121
↓ 36 callersFunctionget_bundle
dnn/src/arm_common/pooling/algo.cpp:18
↓ 36 callersFunctionhash
imperative/src/impl/ops/utility.cpp:224
↓ 36 callersMethodmake
imperative/src/impl/interpreter/stack_manager.h:74
↓ 36 callersMethodremove
(self)
imperative/python/megengine/utils/hook.py:13
↓ 36 callersMethodreserve
Used for initialization
dnn/src/cuda/cutlass/manifest.cpp:49
↓ 36 callersMethodscope
imperative/src/impl/op_def.cpp:131
↓ 35 callersMethodassert_values_in_order
dnn/test/common/small_vector.cpp:199
↓ 35 callersFunctionf
(x)
imperative/python/test/unit/functional/test_elemwise.py:338
↓ 35 callersMethodgen_single_val
dnn/test/common/rng.cpp:214
↓ 35 callersMethodget
imperative/python/src/grad.cpp:139
↓ 35 callersMethodptr
src/core/include/megbrain/tensor.h:383
↓ 35 callersMethodrecord
src/core/impl/comp_node/impl_helper.cpp:5
↓ 35 callersFunctiontree_flatten
r"""Flattens a pytree into a list of values and a :class:`TreeDef` that can be used to reconstruct the pytree.
imperative/python/megengine/traced_module/pytree.py:199
↓ 35 callersMethodvalid
dnn/test/common/workspace_wrapper.h:16
↓ 34 callersFunctionapply
src/jit/impl/halide/ast_hl.cpp:377
↓ 34 callersMethodformat
imperative/src/impl/value.cpp:61
↓ 34 callersMethodget_input_tensor
lite/src/network.cpp:139
↓ 34 callersFunctionget_option
imperative/src/impl/interpreter/option_manager.h:55
↓ 34 callersMethodinsert
imperative/src/impl/op_trait.h:223
↓ 34 callersMethodis_physical_contiguous
dnn/src/common/basic_types.cpp:283
↓ 34 callersMethodname
imperative/python/src/tensor.h:82
↓ 34 callersMethodparam
dnn/src/fallback/conv_bias/im2col/algos.h:60
↓ 34 callersMethodprev_dev_ptr
! * \brief previous dev ptr before deallocating dev_tensor; used for * testing and debugging */
src/core/include/megbrain/graph/var_node.h:365
↓ 34 callersMethodto
r"""Copy self :class:`~.Tensor` to specified device. See :func:`~.copy`
imperative/python/megengine/tensor.py:198
↓ 33 callersFunctionTensorValueLowbit4
dnn/test/common/checker.h:451
↓ 33 callersMethoderase
src/custom/impl/manager.cpp:122
← previousnext →401–500 of 33,912, ranked by callers