MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 13 callersFunctiontester
(ishapes, dtype=None)
imperative/python/test/unit/xla/functional/test_xla_communicate.py:22
↓ 13 callersMethodupdate
update the member from C, this will auto used after slice, share
lite/pylite/megenginelite/tensor.py:343
↓ 13 callersFunctionvmax
dnn/src/arm_common/pooling/pooling_helper.h:714
↓ 13 callersFunctionvset
dnn/src/arm_common/pooling/pooling_helper.h:717
↓ 12 callersFunctionU
dnn/src/naive/rng/opr_impl.cpp:30
↓ 12 callersFunction_
()
imperative/python/test/unit/core/test_autodiff.py:195
↓ 12 callersFunctionaccumulate_2_d_vector
dnn/src/arm_common/conv_bias/int8/channel_wise_kernel.cpp:34
↓ 12 callersMethodadd_layout_constraint
src/core/impl/graph/var_node_mem_mgr.cpp:1233
↓ 12 callersMethodadd_outputs
(self, outputs)
imperative/python/megengine/traced_module/expr.py:133
↓ 12 callersMethodallocated
src/core/impl/graph/var_node_mem_mgr/static_mem_alloc/pushdown.cpp:80
↓ 12 callersFunctionbroadcast_to
r"""Broadcasts a tensor to given shape. Args: inp: input tensor. shape: target shape. Returns: output tensor. E
imperative/python/megengine/functional/tensor.py:543
↓ 12 callersMethodcomp_graph
get the associated ComputingGraph
src/gopt/include/megbrain/gopt/framework.h:50
↓ 12 callersMethodcomp_node_valid
! * \brief whether underlying comp_node is valid */
src/core/include/megbrain/tensor.h:181
↓ 12 callersFunctioncreate_qparams
r""" Args: mode: QuantMode: dtype_meta: Union[str: QuantDtypeMeta]: scale: Tensor: zero_point: Tensor:
imperative/python/megengine/quantization/utils.py:145
↓ 12 callersMethodcreate_unique_name
(self, name: str, node: Any = None)
imperative/python/megengine/traced_module/traced_module.py:403
↓ 12 callersMethoddeduce_layout
dnn/src/common/multi_head_attn/proxy_forward_base.cpp:379
↓ 12 callersMethoddispatch_kern
dnn/src/fallback/convolution/algos.cpp:195
↓ 12 callersMethoddst
dnn/src/fallback/conv_bias/opr_impl.cpp:814
↓ 12 callersMethoddtype
imperative/src/include/megbrain/imperative/transformations/complex.h:36
↓ 12 callersMethodelapsed_time_until
src/core/impl/comp_node/impl_helper.cpp:69
↓ 12 callersMethodensure_seed
dnn/src/cuda/rng/opr_impl.h:96
↓ 12 callersFunctionequal
r"""Element-wise equality comparison. Computes the truth value of :math:`x_i == y_i` for each element :math:`x_i` of the input tensor :math:`x`
imperative/python/megengine/functional/elemwise.py:1836
↓ 12 callersMethodexec
imperative/src/impl/dnn_op_helper.h:245
↓ 12 callersMethodfinished
src/core/impl/comp_node/impl_helper.cpp:13
↓ 12 callersFunctionfunc
(inp, dout)
imperative/python/test/unit/xla/functional/test_xla_indexing.py:26
↓ 12 callersMethodget
(self, block=True, timeout=None)
imperative/python/megengine/data/tools/_queue.py:158
↓ 12 callersMethodget
dnn/src/common/cv/interp_helper.h:150
↓ 12 callersFunctionget_execution_strategy
r"""Returns the execution strategy of :class:`~.module.Conv2d` and :func:`~.matmul` See :func:`~.set_execution_strategy` for possible return valu
imperative/python/megengine/functional/debug_param.py:24
↓ 12 callersFunctionis_const
()
imperative/python/megengine/traced_module/_passes/pattern.py:235
↓ 12 callersMethodload
(cls, opr)
imperative/python/megengine/utils/network_node.py:648
↓ 12 callersFunctionload_library
dnn/atlas-stub/src/libacl_cblas-wrap.h:143
↓ 12 callersMethodload_state_dict
r"""Loads a given dictionary created by :func:`state_dict` into this module. If ``strict`` is ``True``, the keys of :func:`state_dict` must ex
imperative/python/megengine/module/module.py:495
↓ 12 callersMethodlocator
src/core/impl/comp_node/rocm/comp_node.cpp:196
↓ 12 callersMethodlow_bit
dnn/include/megdnn/dtype.h:471
↓ 12 callersFunctionmake_call_rec
src/opr/test/cond.cpp:38
↓ 12 callersFunctionmegcoreDestroyDeviceHandle
dnn/src/common/megcore/public_api/device.cpp:19
↓ 12 callersMethodmultinomial
r"""Random variable with multinomial distribution :math:`\operatorname{Mulitnomial}(p)`. The corresponding probability mass function is
imperative/python/megengine/random/rng.py:537
↓ 12 callersFunctionname
dnn/src/cuda/convolution3d/forward/algo.h:97
↓ 12 callersMethodname
dnn/src/fallback/conv_bias/algos.h:15
↓ 12 callersMethodneg
src/gopt/impl/basic_arith/chain.cpp:230
↓ 12 callersMethodnr_devices
dnn/test/rocm/megcore/fixture.h:9
↓ 12 callersMethodopr
imperative/src/impl/proxy_graph/mini_graph.h:409
↓ 12 callersMethodparam
dnn/src/cuda/convolution/backward_filter/algo.h:128
↓ 12 callersFunctionpowc
src/gopt/test/basic_arith.cpp:21
↓ 12 callersMethodrank
(self)
imperative/python/megengine/distributed/group.py:73
↓ 12 callersMethodrecorder
get the output recorder if this output var is enabled, or a dummmy recorder otherwise
src/opr/impl/loop/impl.h:50
↓ 12 callersFunctionregister_format_rule
imperative/src/impl/transformations/format.cpp:157
↓ 12 callersFunctionrep
dnn/test/x86/elemwise.cpp:17
↓ 12 callersMethodrun
dnn/test/common/svd.cpp:98
↓ 12 callersFunctionrun_test
(func, args, ref_shape, is_trace, sym=False)
imperative/python/test/unit/functional/test_elemwise.py:297
↓ 12 callersMethodset
src/core/include/megbrain/dtype.h:95
↓ 12 callersMethodshape
imperative/src/impl/proxy_graph/mini_graph.h:190
↓ 12 callersFunctionsigmoid
r"""Element-wise `1 / ( 1 + exp( -x ) )`.
imperative/python/megengine/functional/nn.py:851
↓ 12 callersFunctionsqrt
r"""Element-wise :math:`\operatorname{sqrt}(x)` function. Calculates the square root for each element :math:`x_i` of the input tensor :math:`
imperative/python/megengine/functional/elemwise.py:762
↓ 12 callersMethodsrc
dnn/src/fallback/conv_bias/opr_impl.cpp:695
↓ 12 callersFunctionstbi__grow_buffer_unsafe
lite/example/cpp_example/mge/cv/stb_image.h:2218
↓ 12 callersFunctionstbi__zreceive
lite/example/cpp_example/mge/cv/stb_image.h:4543
↓ 12 callersMethodsub
imperative/src/impl/physical_tensor.cpp:753
↓ 12 callersMethodtag
! * \brief get corresponding tag for this tag handler */
src/core/impl/graph/static_infer_impl.h:151
↓ 12 callersMethodto_string
=================== ReformatManager::ReformatKey ====================*/
src/gopt/impl/global_layout_transform/reformat_manager.cpp:67
↓ 12 callersFunctiontranspose
dnn/src/x86/utils.cpp:183
↓ 12 callersFunctionunpack_getitem
(inp, tuple_val, *, allow_newaxis=True)
imperative/python/test/unit/core/test_indexing_op.py:136
↓ 12 callersMethodupdate_on_opr_auto_replace
src/gopt/impl/framework.cpp:219
↓ 12 callersFunctionworker
(data, expect)
imperative/python/test/unit/functional/test_functional_distributed.py:30
↓ 12 callersMethodwrap_safe
(cls, value, node)
imperative/python/megengine/traced_module/node.py:362
↓ 12 callersMethodwrite_pod
imperative/src/include/megbrain/imperative/ops/opr_attr.h:17
↓ 11 callersFunctionLITE_get_tensor_layout
lite/lite-c/src/tensor.cpp:193
↓ 11 callersFunctionMGB_DEFINE_CLS_WITH_SUPER
src/opr/include/megbrain/opr/internal/megdnn_opr_wrapper.h:245
↓ 11 callersMethod_dev_tensor
imperative/python/src/tensor.cpp:734
↓ 11 callersFunction_exclude_from_trace
()
imperative/python/megengine/traced_module/tm_config.py:49
↓ 11 callersMethodadd_axis_cont_inplace
! * \brief add an axis before given *axis*, with shape 1 and contiguous * stride */
dnn/include/megdnn/basic_types.h:288
↓ 11 callersMethodalloc
src/core/impl/utils/mempool.cpp:19
↓ 11 callersMethodapply
dnn/src/fallback/powc/opr_impl.cpp:58
↓ 11 callersFunctionapply_on_var_node
imperative/src/impl/ops/rnn.cpp:9
↓ 11 callersMethodapply_quant_weight
r"""Apply weight's observer and fake_quant from ``qconfig`` on ``target``.
imperative/python/megengine/module/qat/module.py:89
↓ 11 callersFunctionas_device
(obj)
imperative/python/megengine/core/_wrap.py:46
↓ 11 callersMethodclose
(self)
imperative/python/megengine/data/tools/_queue.py:183
↓ 11 callersFunctioncollective_comm
r"""Helper function for applying collective communication functions.
imperative/python/megengine/distributed/functional.py:44
↓ 11 callersMethodcols
dnn/src/common/winograd/winograd_generator.h:83
↓ 11 callersMethodcompile_multi_part
src/core/impl/graph/cg_impl.cpp:455
↓ 11 callersMethodcontext
imperative/src/include/megbrain/imperative/cpp_cupti.h:71
↓ 11 callersMethoddefine_enum_state
Set up thread-local state and return a contextmanager for managing it. Args: name: string, converted to lowercase to define the name o
imperative/python/megengine/xla/lib/config.py:226
↓ 11 callersMethoddevice
imperative/src/impl/value.cpp:48
↓ 11 callersMethoddump_tensor
src/core/impl/utils/debug.cpp:436
↓ 11 callersMethodemplace
imperative/src/include/megbrain/imperative/utils/local_ptr.h:95
↓ 11 callersFunctionfind
src/core/include/megbrain/utils/metahelper.h:279
↓ 11 callersMethodfind_op
dnn/src/cuda/cutlass/operation_table.cpp:154
↓ 11 callersFunctionfmod
Remainder of division. \param x first operand \param y second operand \return remainder of floating point division. template<typename T,typename U> ty
dnn/include/megdnn/dtype/half.hpp:2162
↓ 11 callersFunctionfn
(shape)
imperative/python/test/unit/random/test_rng.py:806
↓ 11 callersFunctionfwd
(x)
imperative/python/test/unit/utils/test_network.py:21
↓ 11 callersMethodgen
dnn/test/cuda/argsort.cpp:72
↓ 11 callersFunctiongen_rand_data
src/cambricon/test/magicmind_runtime_opr.cpp:22
↓ 11 callersFunctiongenerate_transpose_args
dnn/test/common/relayout.cpp:14
↓ 11 callersMethodget_input_name
get the input name by the index in the network Args: index: the index of the input name Returns: th
lite/pylite/megenginelite/network.py:788
↓ 11 callersFunctionget_shape
imperative/python/src/grad_override.cpp:58
↓ 11 callersMethodget_type
src/jit/impl/mlir/mlir_gen.cpp:211
↓ 11 callersFunctionget_var_value
(x)
imperative/python/test/helpers/utils.py:31
↓ 11 callersFunctionget_winograd_benchmark_args
dnn/test/x86/conv_bias.cpp:3021
↓ 11 callersFunctionhandle_bias_and_nonlinear
dnn/src/common/conv_bias.cpp:281
↓ 11 callersMethodhash
imperative/src/impl/ops/opr_attr.cpp:166
← previousnext →1,001–1,100 of 33,912, ranked by callers