Index

_

__version__ (in module megengine)

A

abs() (in module megengine.functional) acos() (in module megengine.functional) acosh() (in module megengine.functional) Adadelta (class in megengine.optimizer) Adagrad (class in megengine.optimizer) Adam (class in megengine.optimizer) AdamW (class in megengine.optimizer) adaptive_avg_pool2d() (in module megengine.functional.nn) adaptive_max_pool2d() (in module megengine.functional.nn) AdaptiveAvgPool2d (class in megengine.module) AdaptiveMaxPool2d (class in megengine.module) add() (in module megengine.functional) add_input() (LiteNetworkIO method) add_input_node() (InternalGraph method) add_output() (LiteNetworkIO method) add_output_node() (InternalGraph method) add_param_group() (Optimizer method) all_gather() (in module megengine.functional.distributed) all_reduce_max() (in module megengine.functional.distributed) all_reduce_min() (in module megengine.functional.distributed) all_reduce_sum() (in module megengine.functional.distributed) all_to_all() (in module megengine.functional.distributed) AllreduceCallback (class in megengine.distributed.helper) Apply (class in megengine.traced_module.expr) apply() (CenterCrop method) (ColorJitter method) (Compose method) (Module method) (RandomCrop method) (RandomHorizontalFlip method) (RandomResize method) (RandomResizedCrop method) (RandomVerticalFlip method) (Resize method) (ShortestEdgeResize method) (VisionTransform method)

apply_batch() (Compose method) (VisionTransform method) apply_easy_quant() (in module megengine.quantization) apply_quant_activation() (QATModule method) apply_quant_bias() (QATModule method) apply_quant_weight() (QATModule method) arange() (in module megengine.functional) argmax() (in module megengine.functional) argmin() (in module megengine.functional) argsort() (in module megengine.functional) ArrayDataset (class in megengine.data.dataset) asin() (in module megengine.functional) asinh() (in module megengine.functional) astype() (Parameter method) (Tensor method) async_exec_level (LiteOptions attribute) async_level (in module megengine.config) async_with_callback() (LiteNetwork method) atan() (in module megengine.functional) atan2() (in module megengine.functional) atanh() (in module megengine.functional) atlas_runtime_opr() (in module megengine.functional.external) AtlasRuntimeSubgraph (class in megengine.module.external) attach() (GradManager method) attached_tensors() (GradManager method) auto_optimize_inference (LiteConfig attribute) autocast (class in megengine.amp) avg_pool2d() (in module megengine.functional.nn) AvgPool2d (class in megengine.module)

B

backend (in module megengine.distributed) (LiteConfig attribute) backward() (Function method) (GradManager method) (GradScaler method) bare_model_cryption_name (LiteConfig property) batch() (MapSampler method) batch_conv_bias_activation() (in module megengine.functional.quantized) batch_norm() (in module megengine.functional.nn) BatchMatMulActivation (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized)

BatchNorm1d (class in megengine.module) BatchNorm2d (class in megengine.module) bcast_list_() (in module megengine.distributed) benchmark_kernel (in module megengine.config) benchmark_with_subprocess (in module megengine.config) beta() (in module megengine.random) (RNG method) binary_cross_entropy() (in module megengine.functional.nn) BrightnessTransform (class in megengine.data.transform) broadcast() (in module megengine.functional.distributed) broadcast_to() (in module megengine.functional) buffers() (Module method)

C

calculate_correct_fan() (in module megengine.module.init) calculate_fan_in_and_fan_out() (in module megengine.module.init) calculate_gain() (in module megengine.module.init) CallFunction (class in megengine.traced_module.expr) CallMethod (class in megengine.traced_module.expr) cambricon_runtime_opr() (in module megengine.functional.external) CambriconRuntimeSubgraph (class in megengine.module.external) cd() (in module megengine.hub.tools) ceil() (in module megengine.functional) CenterCrop (class in megengine.data.transform) check_module_exists() (in module megengine.hub.tools) children() (Module method) CIFAR10 (class in megengine.data.dataset) CIFAR100 (class in megengine.data.dataset) Cityscapes (class in megengine.data.dataset) clear_end_points() (TracedModule method) clear_grad() (Optimizer method) clear_memory_pair() (LiteGlobal static method) clear_watch_points() (TracedModule method) clip() (in module megengine.functional) clip_grad_norm() (in module megengine.optimizer) clip_grad_value() (in module megengine.optimizer) coalesce_free_memory() (in module megengine) COCO (class in megengine.data.dataset) Collator (class in megengine.data) collect() (TensorBatchCollector method) collect_by_ctypes() (TensorBatchCollector method) collect_id() (TensorBatchCollector method) collective_comm() (in module megengine.functional.distributed) ColorJitter (class in megengine.data.transform) comp_node_seq_record_level (LiteOptions attribute) compile() (InternalGraph method) complex() (in module megengine.functional) Compose (class in megengine.data.transform) Concat (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) concat() (in module megengine.functional) ConcatDataset (class in megengine.data.dataset) cond_take() (in module megengine.functional) config_layout (LiteIO attribute)

const_shape (LiteOptions attribute) Constant (class in megengine.traced_module.expr) ContrastTransform (class in megengine.data.transform) Conv1d (class in megengine.module) conv1d() (in module megengine.functional.nn) Conv2d (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) conv2d() (in module megengine.functional.nn) Conv3d (class in megengine.module) conv3d() (in module megengine.functional.nn) conv_bias_activation() (in module megengine.functional.quantized) conv_transpose2d() (in module megengine.functional.nn) (in module megengine.functional.quantized) conv_transpose3d() (in module megengine.functional.nn) ConvBn2d (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) ConvBnRelu2d (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) convert_module_format() (in module megengine.amp) convert_tensor_format() (in module megengine.amp) ConvRelu2d (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) ConvTranspose2d (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) ConvTranspose3d (class in megengine.module) copy() (in module megengine.functional) copy_from() (LiteTensor method) correlation() (in module megengine.functional.nn) cos() (in module megengine.functional) cosh() (in module megengine.functional) create_qparams() (in module megengine.quantization) cross() (in module megengine.functional) cross_entropy() (in module megengine.functional.nn) ctc_loss() (in module megengine.functional.nn) cumsum() (in module megengine.functional) cvt_color() (in module megengine.functional.nn)

D

data_type (LiteLayout attribute) DataLoader (class in megengine.data) Dataset (class in megengine.data.dataset) DEBUG (LiteLogLevel attribute) deformable_conv2d() (in module megengine.functional.nn) deformable_psroi_pooling() (in module megengine.functional.nn) DeformableConv2d (class in megengine.module) DeformablePSROIPooling (class in megengine.module) DequantStub (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) detach() (Parameter method) (Tensor method) deterministic_kernel (in module megengine.config) device (Parameter property) (Tensor property) (TensorNode property) device_id (LiteConfig attribute) (LiteNetwork property) (LiteTensor property) device_type (LiteConfig attribute) (LiteTensor property)

diag() (in module megengine.functional) disable (in module megengine.dtr) disable_fake_quant() (in module megengine.quantization) disable_memory_forwarding (in module megengine.config) disable_observer() (in module megengine.quantization) disable_quantize() (Module method) discrete_input_name (LiteConfig attribute) div() (in module megengine.functional) dmap_callback (Parameter attribute) (Tensor attribute) dot() (in module megengine.functional) Dropout (class in megengine.module) dropout() (in module megengine.functional.nn) DTRConfig (class in megengine.jit) dtype (LiteLayout property) (Parameter property) (Tensor property) (TensorNode property) dump() (trace method) dump_layout_transform_model() (LiteNetwork method) dump_persistent_cache() (LiteGlobal static method) dump_tensorrt_cache() (LiteGlobal static method)

E

Elemwise (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) Embedding (class in megengine.module) embedding() (in module megengine.functional.nn) enable (in module megengine.dtr) enable_cpu_inplace_mode() (LiteNetwork method) enable_debug_log() (in module megengine) enable_fake_quant() (in module megengine.quantization) enable_global_layout_transform() (LiteNetwork method) enable_nchw32 (LiteOptions attribute) enable_nchw4 (LiteOptions attribute) enable_nchw44 (LiteOptions attribute) enable_nchw44_dot (LiteOptions attribute) enable_nchw64 (LiteOptions attribute) enable_nchw88 (LiteOptions attribute) enable_nhwcd4 (LiteOptions attribute) enable_observer() (in module megengine.quantization) enable_profile_performance() (LiteNetwork method) enable_sqrt_sampling (in module megengine.dtr)

enabled (in module megengine.amp) equal() (in module megengine.functional) ERROR (LiteLogLevel attribute) ErrorCode (C++ enum) ErrorCode::LITE_INTERNAL_ERROR (C++ enumerator) ErrorCode::LITE_UNKNOWN_ERROR (C++ enumerator) ErrorCode::OK (C++ enumerator) eval() (InternalGraph method) (Module method) evictee_minimum_size (in module megengine.dtr) eviction_threshold (in module megengine.dtr) exclude_from_trace() (in module megengine.jit) exp() (in module megengine.functional) expand_dims() (in module megengine.functional) expm1() (in module megengine.functional) ExponentialMovingAverageObserver (class in megengine.quantization) exprs() (InternalGraph method) extern_opr_subgraph() (in module megengine.functional.external) ExternOprSubgraph (class in megengine.module.external) extra_configure() (LiteNetwork method) eye() (in module megengine.functional)

F

fake_next_exec (LiteOptions attribute) fake_quant_bias() (in module megengine.quantization) fake_quant_tensor() (in module megengine.quantization) FakeQuantize (class in megengine.quantization) fetch() (GitHTTPSFetcher class method) (GitSSHFetcher class method) FetcherError fill_() (in module megengine.module.init) fill_zero() (LiteTensor method) flatten() (in module megengine.functional) (Parameter method) (Tensor method) (TracedModule method) floor() (in module megengine.functional) floor_div() (in module megengine.functional) force_dynamic_alloc (LiteOptions attribute) force_output_dynamic_alloc (LiteOptions attribute) force_output_use_user_specified_memory (LiteOptions attribute) format (Parameter property) (Tensor property) forward() (Function method) (LiteNetwork method) (PassiveObserver method)

free() (TensorBatchCollector method) from_float_module() (BatchMatMulActivation class method) (Concat class method) (Conv2d class method) (ConvTranspose2d class method) (DequantStub class method) (Elemwise class method) (Linear class method) (QATModule class method) (QuantStub class method) from_pretrained() (Embedding class method) from_qat_module() (BatchMatMulActivation class method) (Concat class method) (Conv2d class method) (ConvTranspose2d class method) (DequantStub class method) (Elemwise class method) (Linear class method) (QuantizedModule class method) (QuantStub class method) full() (in module megengine.functional) full_like() (in module megengine.functional) Function (class in megengine.autodiff) fuse_preprocess (LiteOptions attribute)

G

gamma() (in module megengine.random) (RNG method) gather() (in module megengine.functional) (in module megengine.functional.distributed) GaussianNoise (class in megengine.data.transform) GELU (class in megengine.module) gelu() (in module megengine.functional.nn) get() (TensorBatchCollector method) get_activation_dtype() (QATModule method) get_activation_qparams() (QATModule method) get_all_input_name() (LiteNetwork method) get_all_output_name() (LiteNetwork method) get_allocated_memory() (in module megengine) get_backend() (in module megengine.distributed) get_client() (in module megengine.distributed) get_ctypes_memory() (LiteTensor method) get_cuda_compute_capability() (in module megengine) get_cuda_device_property() (in module megengine) get_cuda_version() (in module megengine) get_cudnn_version() (in module megengine) get_data_by_share() (LiteTensor method) get_default_device() (in module megengine) get_dep_exprs() (InternalGraph method) get_device_count() (in module megengine) (LiteGlobal static method) get_discrete_tensor() (LiteNetwork method) get_expr_by_id() (InternalGraph method) get_function_by_type() (InternalGraph method) get_input_name() (LiteNetwork method) get_io_tensor() (LiteNetwork method) get_logger() (in module megengine) get_lr() (LRScheduler method) (MultiStepLR method) get_max_allocated_memory() (in module megengine) get_max_reserved_memory() (in module megengine)

get_mem_status_bytes() (in module megengine) get_method_by_type() (InternalGraph method) get_mm_server_addr() (in module megengine.distributed) get_module_by_type() (InternalGraph method) get_node_by_id() (InternalGraph method) get_node_by_name() (InternalGraph method) get_output_name() (LiteNetwork method) get_py_server_addr() (in module megengine.distributed) get_rank() (in module megengine.distributed) get_reserved_memory() (in module megengine) get_static_memory_alloc_info() (LiteNetwork method) get_tensor_at() (TensorBatchCollector method) get_tensorrt_version() (in module megengine) get_weight_dtype() (QATModule method) get_weight_qparams() (QATModule method) get_world_size() (in module megengine.distributed) GetAttr (class in megengine.traced_module.expr) GitCheckoutError GitHTTPSFetcher (class in megengine.hub.fetcher) GitPullError GitSSHFetcher (class in megengine.hub.fetcher) grad (Parameter attribute) (Tensor attribute) GradManager (class in megengine.autodiff) GradScaler (class in megengine.amp) graph (TracedModule property) graph() (Parameter method) (Tensor method) graph_opt_level (LiteOptions attribute) GraphOptimizationConfig (class in megengine.jit) greater() (in module megengine.functional) greater_equal() (in module megengine.functional) Group (class in megengine.distributed) group_barrier() (in module megengine.distributed) GroupNorm (class in megengine.module)

H

has_compression (LiteConfig attribute) help() (in module megengine.hub) high_prec_dtype (in module megengine.amp) hinge_loss() (in module megengine.functional.nn)

HistogramObserver (class in megengine.quantization) hsigmoid() (in module megengine.functional.nn) hswish() (in module megengine.functional.nn) HueTransform (class in megengine.data.transform)

I

Identity (class in megengine.module) imag() (in module megengine.functional) ImageFolder (class in megengine.data.dataset) ImageNet (class in megengine.data.dataset) import_module() (in module megengine.hub) indexing_one_hot() (in module megengine.functional.nn) Infinite (class in megengine.data) INFO (LiteLogLevel attribute) init_process_group() (in module megengine.distributed) Input (class in megengine.traced_module.expr) inputs (InternalGraph property) insert_exprs() (InternalGraph method) InstanceNorm (class in megengine.module) InternalGraph (class in megengine.traced_module.traced_module) interpolate() (in module megengine.functional.nn) InvalidGitHost InvalidProtocol

InvalidRepo io_bin_dump() (LiteNetwork method) io_txt_dump() (LiteNetwork method) io_type (LiteIO attribute) is_atlas_available() (in module megengine) is_cambricon_available() (in module megengine) is_continue (LiteTensor property) is_cpu_inplace_mode() (LiteNetwork method) is_cuda_available() (in module megengine) is_distributed() (in module megengine.distributed) is_host (LiteIO attribute) is_pinned_host (LiteTensor property) is_rocm_available() (in module megengine) isinf() (in module megengine.functional) isnan() (in module megengine.functional) item() (Parameter method) (Tensor method)

J

jit_level (LiteOptions attribute)

L

l1_loss() (in module megengine.functional.nn) LAMB (class in megengine.optimizer) LAMBFp16 (class in megengine.optimizer) launcher (class in megengine.distributed) layer_norm() (in module megengine.functional.nn) LayerNorm (class in megengine.module) layout (LiteTensor property) leaky_relu() (in module megengine.functional.nn) LeakyReLU (class in megengine.module) left_shift() (in module megengine.functional) less() (in module megengine.functional) less_equal() (in module megengine.functional) Lighting (class in megengine.data.transform) Linear (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized) linear() (in module megengine.functional.nn) linspace() (in module megengine.functional) list() (in module megengine.hub) lite::Allocator (C++ class) lite::Allocator::allocate (C++ function) lite::Allocator::free (C++ function) lite::Allocator::~Allocator (C++ function) lite::AsyncCallback (C++ type) lite::clear_memory_pair (C++ function) lite::Config (C++ struct) lite::Config::backend (C++ member) lite::Config::bare_model_cryption_name (C++ member) lite::Config::device_id (C++ member) lite::Config::device_type (C++ member) lite::Config::has_compression (C++ member) lite::Config::options (C++ member) lite::DecryptionFunc (C++ type) lite::dump_persistent_cache (C++ function) lite::FinishCallback (C++ type) lite::get_device_count (C++ function) lite::get_log_level (C++ function) lite::get_version (C++ function) lite::IO (C++ struct) lite::IO::config_layout (C++ member) lite::IO::io_type (C++ member) lite::IO::is_host (C++ member) lite::IO::name (C++ member) lite::Layout (C++ struct) lite::Layout::data_type (C++ member) lite::Layout::get_elem_size (C++ function) lite::Layout::MAXDIM (C++ member) lite::Layout::ndim (C++ member) lite::Layout::operator== (C++ function) lite::Layout::shapes (C++ member) lite::lookup_physic_ptr (C++ function) lite::Network (C++ class) lite::Network::compute_only_configured_output (C++ function) lite::Network::enable_profile_performance (C++ function) lite::Network::extra_configure (C++ function) lite::Network::forward (C++ function) lite::Network::get_all_input_name (C++ function) lite::Network::get_all_output_name (C++ function) lite::Network::get_device_id (C++ function) lite::Network::get_device_type (C++ function) lite::Network::get_input_name (C++ function) lite::Network::get_input_tensor (C++ function) lite::Network::get_io_tensor (C++ function) lite::Network::get_model_extra_info (C++ function) lite::Network::get_output_name (C++ function) lite::Network::get_output_tensor (C++ function) lite::Network::get_static_memory_alloc_info (C++ function) lite::Network::get_stream_id (C++ function) lite::Network::load_model (C++ function), [1] lite::Network::Network (C++ function), [1] lite::Network::set_async_callback (C++ function) lite::Network::set_device_id (C++ function) lite::Network::set_finish_callback (C++ function) lite::Network::set_start_callback (C++ function) lite::Network::set_stream_id (C++ function) lite::Network::wait (C++ function) lite::Network::~Network (C++ function) lite::NetworkIO (C++ struct) lite::NetworkIO::inputs (C++ member) lite::NetworkIO::outputs (C++ member) lite::Options (C++ struct) lite::Options::async_exec_level (C++ member) lite::Options::comp_node_seq_record_level (C++ member) lite::Options::const_shape (C++ member) lite::Options::enable_nchw32 (C++ member) lite::Options::enable_nchw4 (C++ member) lite::Options::enable_nchw44 (C++ member) lite::Options::enable_nchw44_dot (C++ member) lite::Options::enable_nchw64 (C++ member) lite::Options::enable_nchw88 (C++ member) lite::Options::enable_nhwcd4 (C++ member) lite::Options::fake_next_exec (C++ member) lite::Options::force_dynamic_alloc (C++ member) lite::Options::force_output_dynamic_alloc (C++ member) lite::Options::force_output_use_user_specified_memory (C++ member) lite::Options::fuse_preprocess (C++ member) lite::Options::graph_opt_level (C++ member) lite::Options::jit_level (C++ member) lite::Options::no_profiling_on_shape_change (C++ member) lite::Options::var_sanity_check_first_run (C++ member) lite::Options::weight_preprocess (C++ member) lite::ParseInfoFunc (C++ type) lite::register_decryption_and_key (C++ function) lite::register_memory_pair (C++ function) lite::register_parse_info_func (C++ function) lite::Runtime (C++ class) lite::Runtime::dump_layout_transform_model (C++ function) lite::Runtime::enable_global_layout_transform (C++ function) lite::Runtime::enable_io_bin_dump (C++ function) lite::Runtime::enable_io_txt_dump (C++ function) lite::Runtime::get_cpu_threads_number (C++ function) lite::Runtime::get_model_io_info (C++ function), [1] lite::Runtime::is_cpu_inplace_mode (C++ function) lite::Runtime::set_cpu_inplace_mode (C++ function) lite::Runtime::set_cpu_threads_number (C++ function) lite::Runtime::set_memory_allocator (C++ function) lite::Runtime::set_network_algo_policy (C++ function) lite::Runtime::set_network_algo_workspace_limit (C++ function) lite::Runtime::set_runtime_thread_affinity (C++ function) lite::Runtime::share_runtime_memory_with (C++ function) lite::Runtime::shared_weight_with_network (C++ function) lite::Runtime::use_tensorrt (C++ function) lite::set_loader_lib_path (C++ function) lite::set_log_level (C++ function) lite::set_persistent_cache (C++ function) lite::StartCallback (C++ type) lite::Tensor (C++ class)

lite::Tensor::copy_from (C++ function) lite::Tensor::fill_zero (C++ function) lite::Tensor::get_device_id (C++ function) lite::Tensor::get_device_type (C++ function) lite::Tensor::get_layout (C++ function) lite::Tensor::get_memory_ptr (C++ function), [1] lite::Tensor::get_tensor_total_size_in_byte (C++ function) lite::Tensor::is_continue_memory (C++ function) lite::Tensor::is_pinned_host (C++ function) lite::Tensor::reset (C++ function), [1] lite::Tensor::reshape (C++ function) lite::Tensor::set_layout (C++ function) lite::Tensor::share_memory_with (C++ function) lite::Tensor::slice (C++ function) lite::Tensor::Tensor (C++ function), [1], [2], [3], [4], [5] lite::Tensor::update_from_implement (C++ function) lite::Tensor::~Tensor (C++ function) lite::ThreadAffinityCallback (C++ type) lite::try_coalesce_all_free_memory (C++ function) lite::update_decryption_or_key (C++ function) LITE_ALGO_HEURISTIC (LiteAlgoSelectStrategy attribute) LITE_ALGO_OPTIMIZED (LiteAlgoSelectStrategy attribute) LITE_ALGO_PROFILE (LiteAlgoSelectStrategy attribute) LITE_ALGO_REPRODUCIBLE (LiteAlgoSelectStrategy attribute) LITE_ATLAS (LiteDeviceType attribute) LITE_CAMBRICON (LiteDeviceType attribute) LITE_CPU (LiteDeviceType attribute) LITE_CUDA (LiteDeviceType attribute) LITE_DEFAULT (LiteBackend attribute) LITE_DEVICE_DEFAULT (LiteDeviceType attribute) LITE_FLOAT (LiteDataType attribute) LITE_HALF (LiteDataType attribute) LITE_INPUT (LiteTensorPhase attribute) LITE_INT (LiteDataType attribute) LITE_INT16 (LiteDataType attribute) LITE_INT8 (LiteDataType attribute) LITE_IO (LiteTensorPhase attribute) LITE_IO_SHAPE (LiteIOType attribute) LITE_IO_VALUE (LiteIOType attribute) LITE_NPU (LiteDeviceType attribute) LITE_OUTPUT (LiteTensorPhase attribute) LITE_UINT16 (LiteDataType attribute) LITE_UINT8 (LiteDataType attribute) LiteAlgoSelectStrategy (C++ enum) (class in megenginelite.struct) LiteAlgoSelectStrategy::LITE_ALGO_HEURISTIC (C++ enumerator) LiteAlgoSelectStrategy::LITE_ALGO_OPTIMIZED (C++ enumerator) LiteAlgoSelectStrategy::LITE_ALGO_PROFILE (C++ enumerator) LiteAlgoSelectStrategy::LITE_ALGO_REPRODUCIBLE (C++ enumerator) LiteBackend (C++ enum) (class in megenginelite.struct) LiteBackend::LITE_DEFAULT (C++ enumerator) LiteConfig (class in megenginelite.network) LiteDataType (C++ enum) (class in megenginelite.struct) LiteDataType::LITE_FLOAT (C++ enumerator) LiteDataType::LITE_HALF (C++ enumerator) LiteDataType::LITE_INT (C++ enumerator) LiteDataType::LITE_INT16 (C++ enumerator) LiteDataType::LITE_INT64 (C++ enumerator) LiteDataType::LITE_INT8 (C++ enumerator) LiteDataType::LITE_UINT (C++ enumerator) LiteDataType::LITE_UINT16 (C++ enumerator) LiteDataType::LITE_UINT8 (C++ enumerator) LiteDeviceType (C++ enum) (class in megenginelite.struct) LiteDeviceType::LITE_ATLAS (C++ enumerator) LiteDeviceType::LITE_CAMBRICON (C++ enumerator) LiteDeviceType::LITE_CPU (C++ enumerator) LiteDeviceType::LITE_CUDA (C++ enumerator) LiteDeviceType::LITE_DEVICE_DEFAULT (C++ enumerator) LiteDeviceType::LITE_NPU (C++ enumerator) LiteGlobal (class in megenginelite.global_setting) LiteIO (class in megenginelite.network) LiteIOType (C++ enum) (class in megenginelite.struct) LiteIOType::LITE_IO_SHAPE (C++ enumerator) LiteIOType::LITE_IO_VALUE (C++ enumerator) LiteLayout (class in megenginelite.tensor) LiteLogLevel (C++ enum) (class in megenginelite.struct) LiteLogLevel::DEBUG (C++ enumerator) LiteLogLevel::ERROR (C++ enumerator) LiteLogLevel::INFO (C++ enumerator) LiteLogLevel::WARN (C++ enumerator) LiteNetwork (class in megenginelite.network) LiteNetworkIO (class in megenginelite.network) LiteOpenCLCacheCompatLevel (C++ enum) LiteOpenCLCacheCompatLevel::LITE_CACHE_COMPAT_LEVEL_CNT (C++ enumerator) LiteOpenCLCacheCompatLevel::LITE_NOT_COMPAT (C++ enumerator) LiteOpenCLCacheCompatLevel::LITE_SERIES_COMPAT (C++ enumerator) LiteOpenCLCacheCompatLevel::LITE_VENDOR_COMPAT (C++ enumerator) LiteOptions (class in megenginelite.network) LiteTensor (class in megenginelite.tensor) LiteTensorPhase (C++ enum) (class in megenginelite.struct) LiteTensorPhase::LITE_INPUT (C++ enumerator) LiteTensorPhase::LITE_IO (C++ enumerator) LiteTensorPhase::LITE_OUTPUT (C++ enumerator) load() (in module megengine) (in module megengine.hub) (LiteNetwork method) load_module() (in module megengine.hub.tools) load_serialized_obj_from_url() (in module megengine.hub) load_state_dict() (LRScheduler method) (Module method) (MultiStepLR method) (Optimizer method) local_conv2d() (in module megengine.functional.nn) local_response_norm() (in module megengine.functional.nn) LocalConv2d (class in megengine.module) LocalResponseNorm (class in megengine.module) log() (in module megengine.functional) log1p() (in module megengine.functional) logaddexp() (in module megengine.functional) logical_and() (in module megengine.functional) logical_not() (in module megengine.functional) logical_or() (in module megengine.functional) logical_xor() (in module megengine.functional) logsigmoid() (in module megengine.functional.nn) logsoftmax() (in module megengine.functional.nn) logsumexp() (in module megengine.functional.nn) lookup_physic_ptr() (LiteGlobal static method) low_prec_dtype (in module megengine.amp) LRScheduler (class in megengine.optimizer) LSQ (class in megengine.quantization) LSTM (class in megengine.module) LSTMCell (class in megengine.module)

M

magicmind_runtime_opr() (in module megengine.functional.external) MagicMindRuntimeSubgraph (class in megengine.module.external) make_allreduce_cb (in module megengine.distributed) MapSampler (class in megengine.data) matinv() (in module megengine.functional) matmul() (in module megengine.functional) max() (in module megengine.functional) (Parameter method) (Tensor method) max_pool2d() (in module megengine.functional.nn) maximum() (in module megengine.functional) MaxPool2d (class in megengine.module) mean() (in module megengine.functional) (Parameter method) (Tensor method) megengine module megengine.amp module megengine.autodiff module megengine.config module megengine.data module megengine.data.collator module megengine.data.dataset module megengine.data.sampler module megengine.data.transform module megengine.distributed module megengine.dtr module megengine.functional module megengine.functional.distributed module megengine.functional.external module megengine.functional.nn module megengine.functional.quantized module megengine.hub module megengine.hub.exceptions module megengine.hub.fetcher module megengine.hub.tools module megengine.jit module megengine.module module megengine.module.external module megengine.module.init module megengine.module.qat module megengine.module.quantized module megengine.optimizer module megengine.quantization module megengine.random module

megengine.traced_module module megengine.traced_module.expr module megengine.traced_module.node module megengine.traced_module.traced_module module megenginelite.global_setting module megenginelite.network module megenginelite.struct module megenginelite.tensor module megenginelite.utils module min() (in module megengine.functional) (Parameter method) (Tensor method) minimum() (in module megengine.functional) MinMaxObserver (class in megengine.quantization) MNIST (class in megengine.data.dataset) mod() (in module megengine.functional) module megengine megengine.amp megengine.autodiff megengine.config megengine.data megengine.data.collator megengine.data.dataset megengine.data.sampler megengine.data.transform megengine.distributed megengine.dtr megengine.functional megengine.functional.distributed megengine.functional.external megengine.functional.nn megengine.functional.quantized megengine.hub megengine.hub.exceptions megengine.hub.fetcher megengine.hub.tools megengine.jit megengine.module megengine.module.external megengine.module.init megengine.module.qat megengine.module.quantized megengine.optimizer megengine.quantization megengine.random megengine.traced_module megengine.traced_module.expr megengine.traced_module.node megengine.traced_module.traced_module megenginelite.global_setting megenginelite.network megenginelite.struct megenginelite.tensor megenginelite.utils Module (class in megengine.module) module_type (ModuleNode attribute) ModuleNode (class in megengine.traced_module.node) modules() (Module method) msra_normal_() (in module megengine.module.init) msra_uniform_() (in module megengine.module.init) mul() (in module megengine.functional) multi_head_attention() (in module megengine.functional.nn) MultiStepLR (class in megengine.optimizer)

N

name (GetAttr attribute) (InternalGraph property) (LiteIO property) (Parameter property) (Tensor property) named_buffers() (Module method) named_children() (Module method) named_modules() (Module method) named_parameters() (Module method) named_tensors() (Module method) nbytes (LiteTensor property) ndim (LiteLayout attribute) (Parameter property) (Tensor property) neg() (in module megengine.functional)

new_group() (in module megengine.distributed) nms() (in module megengine.functional.nn) no_profiling_on_shape_change (LiteOptions attribute) nodes() (InternalGraph method) non_zero() (in module megengine.functional) norm() (in module megengine.functional) normal() (in module megengine.random) (RNG method) normal_() (in module megengine.module.init) Normalize (class in megengine.data.transform) normalize() (in module megengine.functional) not_equal() (in module megengine.functional) numpy() (Parameter method) (Tensor method) nvof() (in module megengine.functional.nn)

O

Objects365 (class in megengine.data.dataset) Observer (class in megengine.quantization) one_hot() (in module megengine.functional.nn) ones() (in module megengine.functional) ones_() (in module megengine.module.init)

ones_like() (in module megengine.functional) Optimizer (class in megengine.optimizer) options (LiteConfig attribute) outputs (InternalGraph property) override_backend() (in module megengine.distributed) owner (ModuleNode property)

P

pack_allreduce_split() (in module megengine.distributed.helper) Pad (class in megengine.data.transform) (class in megengine.module) pad() (in module megengine.functional.nn) param_pack_concat() (in module megengine.distributed.helper) param_pack_split() (in module megengine.distributed.helper) Parameter (class in megengine) parameters() (Module method) PascalVOC (class in megengine.data.dataset) PassiveObserver (class in megengine.quantization) permutation() (in module megengine.random) (RNG method) pixel_shuffle() (in module megengine.functional.nn)

PixelShuffle (class in megengine.module) poisson() (in module megengine.random) (RNG method) polar() (in module megengine.functional) pow() (in module megengine.functional) PReLU (class in megengine.module) prelu() (in module megengine.functional.nn) pretrained (class in megengine.hub) prod() (in module megengine.functional) (Parameter method) (Tensor method) propagate_qconfig() (in module megengine.quantization) PseudoTransform (class in megengine.data.transform)

Q

QATModule (class in megengine.module.qat) QConfig (class in megengine.quantization) QParams (class in megengine.quantization) qparams (Parameter property) (Tensor property) (TensorNode property) qualname (InternalGraph property)

quantize() (in module megengine.quantization) quantize_qat() (in module megengine.quantization) QuantizedModule (class in megengine.module.quantized) QuantMode (class in megengine.quantization) QuantStub (class in megengine.module) (class in megengine.module.qat) (class in megengine.module.quantized)

R

RandomCrop (class in megengine.data.transform) RandomHorizontalFlip (class in megengine.data.transform) RandomResize (class in megengine.data.transform) RandomResizedCrop (class in megengine.data.transform) RandomSampler (class in megengine.data) RandomVerticalFlip (class in megengine.data.transform) real() (in module megengine.functional) record() (GradManager method) reduce_scatter_sum() (in module megengine.functional.distributed) reduce_sum() (in module megengine.functional.distributed) register_as_builtin() (in module megengine.traced_module) register_decryption_and_key() (LiteGlobal static method) register_forward_hook() (Module method) register_forward_pre_hook() (Module method) register_memory_pair() (LiteGlobal static method) release() (GradManager method) ReLU (class in megengine.module) relu() (in module megengine.functional.nn) relu6() (in module megengine.functional.nn) remap() (in module megengine.functional.nn) remote_recv() (in module megengine.functional.distributed) remote_send() (in module megengine.functional.distributed) repeat() (in module megengine.functional)

replace_node() (InternalGraph method) replace_param() (Module method) ReplacementSampler (class in megengine.data) requires_grad (Parameter property) (Tensor property) reset_max_memory_stats() (in module megengine) reset_outputs() (InternalGraph method) reset_qconfig() (in module megengine.quantization) reset_zero() (Parameter method) (Tensor method) reshape() (in module megengine.functional) (LiteTensor method) (Parameter method) (Tensor method) Resize (class in megengine.data.transform) right_shift() (in module megengine.functional) RNG (class in megengine.random) RNN (class in megengine.module) RNNCell (class in megengine.module) roi_align() (in module megengine.functional.nn) roi_pooling() (in module megengine.functional.nn) roll() (in module megengine.functional) round() (in module megengine.functional)

S

sample() (Infinite method) (MapSampler method) (RandomSampler method) (ReplacementSampler method) (SequentialSampler method) Sampler (class in megengine.data) SaturationTransform (class in megengine.data.transform) save() (in module megengine) scatter() (in module megengine.functional) (in module megengine.functional.distributed) (MapSampler method) seed() (in module megengine.random) Sequential (class in megengine.module) SequentialSampler (class in megengine.data) Server (class in megengine.distributed) set_data_by_copy() (LiteTensor method) set_data_by_share() (LiteTensor method) set_default_device() (in module megengine) set_end_points() (TracedModule method) set_finish_callback() (LiteNetwork method) set_loader_lib_path() (LiteGlobal static method) set_log_file() (in module megengine) set_log_level() (in module megengine) set_network_algo_policy() (LiteNetwork method) set_network_algo_workspace_limit() (LiteNetwork method) set_persistent_cache() (LiteGlobal static method) set_prealloc_config() (in module megengine) set_qconfig() (QATModule method) set_qparams() (PassiveObserver method) set_start_callback() (LiteNetwork method) set_tensorrt_cache() (LiteGlobal static method) set_value() (Parameter method) (Tensor method) set_watch_points() (TracedModule method) SGD (class in megengine.optimizer) shape (Parameter property) (Tensor property) (TensorNode property) shapes (LiteLayout property) share_memory_with() (LiteTensor method) share_runtime_memroy() (LiteNetwork method) share_weights_with() (LiteNetwork method) ShortestEdgeResize (class in megengine.data.transform) shuffle() (in module megengine.random) (RNG method)

Sigmoid (class in megengine.module) sigmoid() (in module megengine.functional.nn) sign() (in module megengine.functional) SiLU (class in megengine.module) silu() (in module megengine.functional.nn) sin() (in module megengine.functional) sinh() (in module megengine.functional) size (Parameter property) (Tensor property) slice() (LiteTensor method) sliding_window() (in module megengine.functional.nn) sliding_window_transpose() (in module megengine.functional.nn) SlidingWindow (class in megengine.module) SlidingWindowTranspose (class in megengine.module) Softmax (class in megengine.module) softmax() (in module megengine.functional.nn) softplus() (in module megengine.functional.nn) sort() (in module megengine.functional) split() (in module megengine.functional) sqrt() (in module megengine.functional) square() (in module megengine.functional) square_loss() (in module megengine.functional.nn) squeeze() (in module megengine.functional) stack() (in module megengine.functional) state_dict() (LRScheduler method) (Module method) (MultiStepLR method) (Optimizer method) std() (in module megengine.functional) step() (Optimizer method) stream_id (LiteNetwork property) StreamDataset (class in megengine.data.dataset) StreamSampler (class in megengine.data) sub() (in module megengine.functional) SublinearMemoryConfig (class in megengine.jit) sum() (in module megengine.functional) (Parameter method) (Tensor method) svd() (in module megengine.functional) swapaxes() (in module megengine.functional) sync_batch_norm() (in module megengine.functional.nn) SyncBatchNorm (class in megengine.module) SyncExponentialMovingAverageObserver (class in megengine.quantization) synchronized() (in module megengine.distributed) (in module megengine.hub.fetcher) SyncMinMaxObserver (class in megengine.quantization)

T

T (Parameter property) (Tensor property) tan() (in module megengine.functional) tanh() (in module megengine.functional) Tensor (class in megengine) TensorBatchCollector (class in megenginelite.utils) TensorNode (class in megengine.traced_module.node) tensorrt_runtime_opr() (in module megengine.functional.external) TensorrtRuntimeSubgraph (class in megengine.module.external) tensors() (Module method) threads_number (LiteNetwork property) tile() (in module megengine.functional) to() (Parameter method) (Tensor method) to_numpy() (LiteTensor method) (TensorBatchCollector method) tolist() (Parameter method) (Tensor method)

ToMode (class in megengine.data.transform) top_graph (InternalGraph property) topk() (in module megengine.functional) topk_accuracy() (in module megengine.functional) TorchTransformCompose (class in megengine.data.transform) TQT (class in megengine.quantization) trace (class in megengine.jit) trace_module() (in module megengine.traced_module) TracedModule (class in megengine.traced_module) train() (Module method) (Observer method) Transform (class in megengine.data.transform) transpose() (in module megengine.functional) (Parameter method) (Tensor method) try_coalesce_all_free_memory() (LiteGlobal static method)

U

uniform() (in module megengine.random) (RNG method) uniform_() (in module megengine.module.init) unscale() (GradScaler method)

update() (GradScaler method) (LiteTensor method) update_decryption_key() (LiteGlobal static method) use_tensorrt() (LiteNetwork method)

V

value (Constant attribute) (TensorNode property) value_id() (Parameter method) (Tensor method) var() (in module megengine.functional) (Parameter method) (Tensor method)

var_sanity_check_first_run (LiteOptions attribute) VisionDataset (class in megengine.data.dataset) VisionTransform (class in megengine.data.transform)

W

wait() (LiteNetwork method) WARN (LiteLogLevel attribute) warp_affine() (in module megengine.functional.nn) warp_perspective() (in module megengine.functional.nn)

weight_preprocess (LiteOptions attribute) what_is_xpu() (in module megengine.device) where() (in module megengine.functional) wrap() (in module megengine.traced_module)

X

xavier_normal_() (in module megengine.module.init)

xavier_uniform_() (in module megengine.module.init) xla_trace (class in megengine.jit)

Z

zero_grad() (Module method) (Optimizer method)

zeros() (in module megengine.functional) zeros_() (in module megengine.module.init) zeros_like() (in module megengine.functional)