MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / functions

Functions33,912 in github.com/MegEngine/MegEngine

↓ 8 callersMethodstr
lite/load_and_run/src/helpers/json_loader.cpp:67
↓ 8 callersMethodto_contiguous_inplace
imperative/src/impl/physical_tensor.cpp:656
↓ 8 callersMethodtolist
r"""Returns the tensor as a (nested) list. For scalars, a standard Python number is returned, just like with :meth:`~.item`. Tensors a
imperative/python/megengine/core/tensor/array_method.py:381
↓ 8 callersFunctiontrans
src/opr/impl/blas.cpp:609
↓ 8 callersFunctiontranspose_4x2x8_int8_int16
dnn/src/x86/conv_bias/int8/common_helper.h:56
↓ 8 callersFunctiontranspose_4x2xn_int8_int16
dnn/src/x86/conv_bias/int8/common_helper.h:39
↓ 8 callersFunctiontranspose_4x4_1_s
dnn/src/aarch64/matrix_mul/asm/common.h:1535
↓ 8 callersMethodtype
dnn/include/megdnn/tensor_format.h:46
↓ 8 callersMethodvalid
src/plugin/include/megbrain/plugin/static_mem_record.h:28
↓ 8 callersFunctionvmovl_high_s8
armv7 : vmovl_xx(vget_high_xx()), armv8 : vmovl_high_xx()
dnn/src/arm_common/simd_macro/marm_neon.h:370
↓ 8 callersFunctionvmovl_low_s8
pack vmovl_low_xx() on armv7 and armv8
dnn/src/arm_common/simd_macro/marm_neon.h:419
↓ 8 callersFunctionvst1q_f32_x2
dnn/src/arm_common/simd_macro/marm_neon.h:307
↓ 8 callersFunctionxla_scatter
This is xla Scatter op wrapper, which is inverse process of Gather. Please read the docstring of Gather op before reading the below contents.
imperative/python/megengine/xla/rules/tensor.py:518
↓ 7 callersFunctionConcatFile
( split_number: int, file_path: str, operations: str, type: str, head: str, required_c
dnn/scripts/cutlass_generator/generator.py:1898
↓ 7 callersMethodGetHandle
src/opr/impl/nvof/NvOFCuda.h:49
↓ 7 callersFunctionGiBlendInt8
dnn/src/fallback/general_intrinsic/gi_int.h:664
↓ 7 callersFunctionLITE_tensor_copy
lite/lite-c/src/tensor.cpp:149
↓ 7 callersFunctionLITE_wait
lite/lite-c/src/network.cpp:265
↓ 7 callersMethodSubtensor
src/opr/impl/tensor_manip.cpp:626
↓ 7 callersMethod__init__
(self)
imperative/python/test/unit/utils/test_module_stats.py:110
↓ 7 callersFunction_graph_surgery_mode
()
imperative/python/megengine/traced_module/tm_config.py:37
↓ 7 callersFunction_init_module
()
imperative/python/test/unit/traced_module/test_modification.py:104
↓ 7 callersMethod_process
(self, defs)
dnn/scripts/gen_param_defs.py:314
↓ 7 callersMethod_read
(self, name)
imperative/python/megengine/xla/lib/config.py:88
↓ 7 callersMethod_write_def
(self, name, val)
tools/gen_header_for_bin_reduce.py:225
↓ 7 callersMethod_write_doc
(self, doc)
dnn/scripts/gen_param_defs.py:479
↓ 7 callersFunctionacos
Arc cosine function. \param arg function argument \return arc cosine value of \a arg template<typename T> typename enable<expr,T>::type acos(T arg) {
dnn/include/megdnn/dtype/half.hpp:2365
↓ 7 callersMethodadd_elemwise
src/gopt/test/network.cpp:106
↓ 7 callersMethodadd_input
add input information into LiteNetworkIO
lite/pylite/megenginelite/network.py:423
↓ 7 callersMethodadd_opr
src/gopt/impl/framework.cpp:861
↓ 7 callersMethodadd_output
r"""Adds vars into the network output node list
imperative/python/megengine/utils/network.py:282
↓ 7 callersMethodadd_output
src/core/impl/graph/operator_node.cpp:240
↓ 7 callersMethodadd_pooling
src/gopt/test/network.cpp:122
↓ 7 callersMethodadd_text
Add text data to summary. Args: tag (string): Data identifier text_string_list (string list): String to save
imperative/python/megengine/utils/tensorboard.py:142
↓ 7 callersMethodalignment
! * \brief get current alignment */
src/core/include/megbrain/comp_node/alloc.h:263
↓ 7 callersMethodalignment_requirement
dnn/src/cuda/handle.cpp:111
↓ 7 callersMethodall_oprs
(self)
imperative/python/megengine/utils/network.py:479
↓ 7 callersMethodalloc
src/core/impl/comp_node/mem_alloc/impl.cpp:239
↓ 7 callersMethodapply
(self, inputs)
imperative/python/megengine/data/collator.py:33
↓ 7 callersMethodas
imperative/src/include/megbrain/imperative/value.h:402
↓ 7 callersMethodas_count
r"""consume this iterator and get the number of elements
imperative/python/megengine/utils/network.py:751
↓ 7 callersMethodas_fwd_args
dnn/src/rocm/convolution/backward_data/algo.h:44
↓ 7 callersMethodattached_tensors
r"""Return attached tensor list from :meth:`attach`.
imperative/python/megengine/autodiff/grad_manager.py:127
↓ 7 callersFunctionbenchmark_target_algo
dnn/test/cuda/batch_conv_bias.cpp:196
↓ 7 callersFunctionborder_interpolate
dnn/src/common/cv/helper.h:236
↓ 7 callersFunctionbroadcast_to
[x, y, z] or [x, 1, z] only can broadcast to [a, b, c, ..., x, y, z], rather than [x, y, z, ..., a, b, c] but you can realize the latter spec
imperative/python/megengine/xla/rules/tensor.py:19
↓ 7 callersFunctionbuiltin_isinf
dnn/include/megdnn/dtype/half.hpp:257
↓ 7 callersFunctioncan_use_mha_cudnn
dnn/src/cuda/multi_head_attn/opr_impl.cpp:10
↓ 7 callersMethodceil
dnn/include/megdnn/dtype/half.hpp:1579
↓ 7 callersMethodcheck_async_error
src/core/impl/comp_node/comp_node.cpp:478
↓ 7 callersMethodcheck_exception
src/core/impl/utils/async_worker.cpp:117
↓ 7 callersFunctioncheck_reproducibility
src/opr/test/rand.cpp:32
↓ 7 callersFunctioncheck_wait
src/core/test/graph/misc.cpp:1074
↓ 7 callersFunctionchecker_conv_bias
dnn/test/x86/conv_bias.cpp:1437
↓ 7 callersFunctionchecker_conv_bias_int8x8x16
dnn/test/common/conv_bias.cpp:1447
↓ 7 callersMethodclear_device_memory
src/core/impl/graph/cg_impl.cpp:824
↓ 7 callersMethodconsumer_commit
src/core/impl/utils/thread.cpp:230
↓ 7 callersFunctionconvert_to_clayout
convert the lite::Layout to Layout
lite/lite-c/src/tensor.cpp:33
↓ 7 callersFunctionconvert_to_conv_param
src/opr/test/dnn/convolution.cpp:282
↓ 7 callersFunctioncos
r"""Element-wise :math:`\cos(x)` function. Calculates an approximation to the cosine for each element :math:`x_i` of the input tensor :math:`
imperative/python/megengine/functional/elemwise.py:1088
↓ 7 callersFunctioncreate_quantized_dtype
r"""Get quantized dtype with metadata attribute according to _metadata_dict. Note that unsigned dtype must have ``zero_point`` and signed dtype m
imperative/python/megengine/core/tensor/dtype.py:114
↓ 7 callersFunctioncvt_src_or_dst_nchw2nhwc
dnn/test/common/utils.h:227
↓ 7 callersMethoddeactive
src/core/impl/utils/thread_pool.cpp:139
↓ 7 callersFunctiondeduce_reformat_layout
dnn/src/cuda/conv_bias/conv_nchwqs8.cpp:13
↓ 7 callersFunctiondemangle
(name)
scripts/gdb/commands.py:10
↓ 7 callersMethoddetach
imperative/python/src/tensor.cpp:729
↓ 7 callersMethoddisable_freelist
! * \brief disable free list for memory reuse * * This is only useful in the destructor of an enclosing object, so * no extra memo
src/core/include/megbrain/utils/mempool.h:105
↓ 7 callersFunctiondlerror
src/custom/impl/manager.cpp:29
↓ 7 callersFunctiondlsym
lite/src/misc.h:235
↓ 7 callersMethoddtype
imperative/src/impl/proxy_graph/mini_graph.h:63
↓ 7 callersFunctiondtype_np2mgb_descr
imperative/python/src/helper.cpp:312
↓ 7 callersMethoddump_cache
src/tensorrt/impl/tensorrt_engine_cache.cpp:177
↓ 7 callersMethodenable
src/opr/impl/loop/impl.cpp:1009
↓ 7 callersMethodend
src/core/include/megbrain/utils/thin/hash_table.h:344
↓ 7 callersMethodexec
dnn/test/cuda/chanwise_convolution3d.cpp:122
↓ 7 callersFunctionexpand_dhw
(x)
imperative/python/megengine/functional/nn.py:124
↓ 7 callersFunctionfloor
r"""Element-wise :math:`\lfloor x \rfloor` function. Rounds each element :math:`x_i` of the input tensor :math:`x` to the greatest integer-va
imperative/python/megengine/functional/elemwise.py:928
↓ 7 callersMethodformat
(self, record)
imperative/python/megengine/logger.py:62
↓ 7 callersMethodfrom_qat_module
r""" Return a :class:`~.QuantizedModule` instance converted from a :class:`~.QATModule` instance.
imperative/python/megengine/module/quantized/conv.py:90
↓ 7 callersFunctiongcd
src/gopt/impl/global_layout_transform/reformat_manager.cpp:12
↓ 7 callersMethodgen
dnn/test/rocm/argsort.cpp:72
↓ 7 callersMethodgen
dnn/test/common/rng.cpp:87
↓ 7 callersFunctiongenerate
(modes, support_dtypes, output, cpp_ext)
dnn/scripts/gen_elemwise_multi_type_kern_impls.py:16
↓ 7 callersMethodget
src/jit/impl/mlir/mlir_gen.cpp:215
↓ 7 callersMethodget_all_algorithms_info_safe
dnn/include/megdnn/oprs/base.h:356
↓ 7 callersFunctionget_all_algorithms_safe
dnn/src/common/algo_chooser.h:84
↓ 7 callersFunctionget_backend
(platform=None)
imperative/python/megengine/xla/lib/xla_bridge.py:381
↓ 7 callersMethodget_cache
will make m_cache empty
src/core/include/megbrain/utils/persistent_cache.h:56
↓ 7 callersMethodget_default
get default index; fallback to src index if default is not set
dnn/scripts/gen_param_defs.py:219
↓ 7 callersMethodget_inner_block_size
dnn/src/fallback/matrix_mul/opr_impl.h:222
↓ 7 callersFunctionget_int8_chwn4_args_check_bounds
dnn/test/common/conv_bias.cpp:684
↓ 7 callersFunctionget_library_handle
dnn/cuda-stub/src/dlopen_helper.h:69
↓ 7 callersFunctionget_matmul_kern_param
dnn/src/fallback/conv_bias/im2col/algos.cpp:20
↓ 7 callersFunctionget_mm_server_addr
r"""Get master_ip and port of C++ mm_server.
imperative/python/megengine/distributed/group.py:228
↓ 7 callersFunctionget_resnet50_bench_args
dnn/test/cuda/conv_test_utils.cpp:23
↓ 7 callersMethodget_secs_reset
get seconds and reset
src/core/include/megbrain/utils/timer.h:39
↓ 7 callersMethodget_serialized_model
src/cambricon/test/cambricon_runtime_opr.cpp:213
↓ 7 callersMethodget_subopr_list
dnn/src/cuda/conv_bias/matmul.cpp:47
↓ 7 callersMethodget_type
lite/load_and_run/src/helpers/common.h:161
↓ 7 callersMethodget_value
(self)
imperative/python/megengine/core/tensor/megbrain_graph.py:634
← previousnext →1,401–1,500 of 33,912, ranked by callers