MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 26 callersMethodrun
r""" Args: inp_args: list of input datas. inp_dict: dict of named input datas. Returns: a dict {
imperative/python/megengine/utils/comp_graph_tools.py:465
↓ 26 callersMethodset
dnn/src/cuda/dropout/opr_impl.h:13
↓ 26 callersMethodset
dnn/src/cuda/convolution/helper.h:33
↓ 26 callersFunctionsizeof_fmt
(num, suffix="B")
imperative/python/megengine/utils/module_stats.py:198
↓ 26 callersFunctiontester
(OptCls, **kwargs)
imperative/python/test/unit/xla/test_xla_training.py:156
↓ 25 callersFunction_compare_nchw_nhwc
(data, func, is_symbolic=None)
imperative/python/test/unit/core/test_formatted_tensor.py:44
↓ 25 callersMethodbackward
(self, dy)
imperative/python/test/unit/autodiff/test_grad_manager.py:436
↓ 25 callersMethodcode_gen
src/jit/include/megbrain/jit/ast_c.h:55
↓ 25 callersMethoddefine_bool_state
Set up thread-local state and return a contextmanager for managing it. This function is a convenience wrapper. It defines a flag, environment
imperative/python/megengine/xla/lib/config.py:141
↓ 25 callersMethoddispatch_on_comp_node
src/core/impl/graph/normal_exec_env.cpp:104
↓ 25 callersMethodget_used_memory
src/core/impl/comp_node/cuda/comp_node.cpp:537
↓ 25 callersMethodget_var
src/jit/impl/executor_opr.cpp:440
↓ 25 callersMethodidx
get logical index; valonly must be false
dnn/include/megdnn/tensor_iter.h:135
↓ 25 callersFunctionincr_refp
dnn/src/common/utils.h:361
↓ 25 callersFunctioninterleave_helper
dnn/src/aarch64/matrix_mul/asm/common.h:1272
↓ 25 callersFunctionis_static_var_value
! * \brief whether var value could be statically inferred */
src/core/include/megbrain/graph/helper.h:40
↓ 25 callersMethoditem
r"""Returns the value of this :class:`~.Tensor` as a standard Python :class:`numbers.Number`. This only works for tensors with one element. Fo
imperative/python/megengine/core/tensor/array_method.py:371
↓ 25 callersMethodname
imperative/src/impl/ops/custom_opdef.cpp:41
↓ 25 callersMethodname
! * \brief name of this data type */
dnn/include/megdnn/dtype.h:448
↓ 25 callersMethodremove_axis
dnn/src/common/basic_types.cpp:245
↓ 25 callersFunctionrun_test
src/opr/test/internal.cpp:53
↓ 25 callersMethodset
dnn/src/rocm/convolution/helper.cpp:31
↓ 25 callersMethodset_data_by_share
share the data to the tensor Args: data: the data will shared to the tensor, it should be a numpy.ndarra
lite/pylite/megenginelite/tensor.py:420
↓ 25 callersMethodsync
imperative/src/impl/persistent_cache.cpp:121
↓ 25 callersMethodtensors
r""" Returns an iterable for the :class:`~.Tensor` of the module. :param recursive: If ``True``, returns all :class:`~.Tensor` within
imperative/python/megengine/module/module.py:300
↓ 25 callersFunctionupdate
dnn/src/naive/lamb/opr_impl.cpp:15
↓ 24 callersFunctionSTBI__CASE
lite/example/cpp_example/mge/cv/stb_image.h:5140
↓ 24 callersFunctionaccumulate_2_q_vector
dnn/src/arm_common/conv_bias/int8/channel_wise_kernel.cpp:11
↓ 24 callersFunctionbenchmark_impl
dnn/test/x86/conv_bias.cpp:2216
↓ 24 callersMethoddetach
r"""Returns a new :class:`~.Tensor`, detached from the current graph.
imperative/python/megengine/tensor.py:157
↓ 24 callersMethoddtype
imperative/python/src/tensor.cpp:684
↓ 24 callersFunctionenable_miopen_algo_search
dnn/src/rocm/utils.h:24
↓ 24 callersMethodexp
dnn/include/megdnn/dtype/half.hpp:1193
↓ 24 callersMethodget
src/tensorrt/impl/tensorrt_engine_cache.cpp:198
↓ 24 callersFunctionget_cv_args
dnn/test/common/resize.h:156
↓ 24 callersFunctionget_default_device
r"""Gets default computing node. It returns the value set by :func:`~.set_default_device`.
imperative/python/megengine/device.py:147
↓ 24 callersMethodinsert
(self, idx, mod: Module)
imperative/python/megengine/traced_module/utils.py:242
↓ 24 callersFunctionisfinite
Check if finite number. \param arg number to check \retval true if neither infinity nor NaN \retval false else template<typename T> typename enable<bo
dnn/include/megdnn/dtype/half.hpp:2640
↓ 24 callersMethodproxy_to_default_cpu
src/core/include/megbrain/tensor.h:545
↓ 24 callersFunctionpy_apply
imperative/python/src/tensor.cpp:82
↓ 24 callersFunctionrep
dnn/test/common/conv_bias.cpp:1166
↓ 24 callersMethodreset
imperative/src/impl/proxy_graph/mini_graph.h:138
↓ 24 callersFunctionsc
dnn/src/cuda/cudnn_wrapper.cpp:329
↓ 24 callersMethodshape
imperative/python/src/tensor.cpp:671
↓ 24 callersFunctiontester
(x_shape, normalized_shape, w_shape, b_shape, eps=1e-5, dtype=None)
imperative/python/test/unit/xla/functional/test_xla_normalize.py:19
↓ 24 callersFunctionvar_from_tensor_shape
! * \brief get a symbolvar whose value is tensor shape * * \param inp used to determine the computing graph, which can be any symbolvar * bel
src/core/include/megbrain/graph/helper.h:254
↓ 24 callersMethodwrap_outputs
imperative/src/impl/transformations/format.cpp:91
↓ 23 callersMethodElemwise
src/opr/impl/basic_arith.cpp:66
↓ 23 callersFunctionapply
imperative/src/include/megbrain/imperative/dispatch.h:40
↓ 23 callersMethodargs
r"""Get the positional arguments of the operation corresponding to this Expr.
imperative/python/megengine/traced_module/expr.py:222
↓ 23 callersFunctioncheck_winograd
dnn/test/common/conv_bias.cpp:1215
↓ 23 callersFunctionconvert_to_cnnl_datatype
dnn/src/cambricon/cnnl_wrapper/cnnl_types.cpp:7
↓ 23 callersMethodget_input_layout_constraint
imperative/src/impl/op_def.cpp:52
↓ 23 callersFunctioninterleave_4
dnn/src/armv7/matrix_mul/asm/common.h:618
↓ 23 callersMethodis_empty
dnn/src/common/basic_types.cpp:177
↓ 23 callersFunctionmegcoreCreateDeviceHandle
dnn/src/common/megcore/public_api/device.cpp:9
↓ 23 callersFunctionmegcoreMemcpy
dnn/src/common/megcore/public_api/computing.cpp:39
↓ 23 callersFunctionmodify_opr_algo_strategy_inplace
r"""C++ graph version of :func:`~.set_execution_strategy`. Used to inplacely modify dumped graph's fast-run strategy. Args: dest_vars
imperative/python/megengine/core/tensor/megbrain_graph.py:343
↓ 23 callersFunctionmul
(inputs, f, c)
imperative/python/test/unit/core/test_subgraph.py:129
↓ 23 callersMethodset_data_by_copy
copy the data to the tensor Args: data: the data to copy to tensor, it should be list, numpy.ndarraya or
lite/pylite/megenginelite/tensor.py:453
↓ 22 callersFunction_mm_cvtepi8_epi16_from_ptr
dnn/src/x86/matrix_mul/common/common.h:537
↓ 22 callersMethodas_ref
imperative/src/include/megbrain/imperative/value.h:203
↓ 22 callersFunctioncnnl_handle
dnn/src/cambricon/utils.h:21
↓ 22 callersFunctionget_conv_bias_args
dnn/test/common/conv_bias.cpp:1272
↓ 22 callersMethodget_single_comp_node
src/core/impl/graph/operator_node.cpp:487
↓ 22 callersMethodget_workspace_size
dnn/src/aarch64/conv_bias/int8/strategy.cpp:210
↓ 22 callersMethodgraph_compile
src/serialization/impl/serializer.cpp:29
↓ 22 callersFunctioninterleave_helper
dnn/src/armv7/matrix_mul/asm/common.h:586
↓ 22 callersFunctionis_profiling
()
imperative/python/megengine/utils/profiler.py:322
↓ 22 callersMethodlocator
get the physical locator that created this comp node
src/core/include/megbrain/comp_node.h:445
↓ 22 callersFunctionmake
imperative/src/include/megbrain/imperative/opr_utility.h:60
↓ 22 callersFunctionmerge_one_out
src/opr/test/cond.cpp:45
↓ 22 callersFunctionprefetch_3x
dnn/src/armv7/matrix_mul/asm/common.h:54
↓ 22 callersMethodreset
imperative/src/impl/proxy_graph.cpp:431
↓ 22 callersMethodreset
src/core/impl/graph/var_node.cpp:13
↓ 22 callersFunctionsetup_algo
imperative/src/impl/algo_chooser.h:10
↓ 22 callersMethodusable_output
src/core/impl/graph/operator_node.cpp:167
↓ 21 callersFunctionasync_error_info
dnn/src/cuda/utils.h:33
↓ 21 callersFunctioncb1
(_, grad)
imperative/python/test/unit/xla/test_xla_training.py:312
↓ 21 callersFunctioncheck_compute_capability
dnn/test/cuda/utils.cpp:5
↓ 21 callersFunctionchecker_conv_bias_qint8x8x8
dnn/test/arm_common/conv_bias_multi_thread.cpp:290
↓ 21 callersMethodcomp_node
src/tensorrt/include/megbrain/tensorrt/tensorrt_opr.h:223
↓ 21 callersMethoddequantize
dnn/include/megdnn/dtype.h:854
↓ 21 callersMethoddigest
src/core/impl/utils/hash.cpp:22
↓ 21 callersMethodexec
dnn/src/common/powc.cpp:8
↓ 21 callersMethodget
imperative/src/include/megbrain/imperative/utils/local_ptr.h:210
↓ 21 callersFunctionget_conv_bias_1x1_args
dnn/test/common/conv_bias.cpp:1356
↓ 21 callersMethodget_scale
dnn/src/common/utils.cpp:239
↓ 21 callersMethodhandle
imperative/src/include/megbrain/imperative/transformations/eval.h:25
↓ 21 callersMethodimpl
src/core/include/megbrain/utils/thin/hash_table.h:314
↓ 21 callersFunctioninit_output_static_infer_desc
src/opr/impl/loop/impl.cpp:163
↓ 21 callersFunctioninterleave_4
dnn/src/aarch64/matrix_mul/asm/common.h:1293
↓ 21 callersMethodload
src/opr/impl/dnn/dnn.sereg.h:212
↓ 21 callersMethodlock
imperative/src/impl/value.cpp:145
↓ 21 callersFunctionmegcoreGetDeviceHandle
dnn/src/common/megcore/public_api/computing.cpp:25
↓ 21 callersMethodoffset
imperative/src/include/megbrain/imperative/physical_tensor.h:140
↓ 21 callersMethodquantize
dnn/include/megdnn/dtype.h:845
↓ 21 callersMethodselect
src/jit/impl/mlir/ir/common.cpp:98
↓ 21 callersFunctionset_symbolic_shape
r"""Set the VarNode use symbolic shape or not, return the last status. Please set to True and must recover after dump if want to change the input
imperative/python/megengine/utils/network.py:535
↓ 21 callersFunctiontensor_formats_to_named_tensor_shape
src/gopt/impl/global_layout_transform/utils.h:97
← previousnext →601–700 of 33,912, ranked by callers