索引

_

__init__() (torch_tensorrt.Device 方法) (torch_tensorrt.Input 方法) (torch_tensorrt.MutableTorchTensorRTModule 方法) (torch_tensorrt.runtime.PythonTorchTensorRTModule 方法) (torch_tensorrt.runtime.TorchTensorRTModule 方法)

B

b (torch_tensorrt.dtype 属性)

bf16 (torch_tensorrt.dtype 属性)

C

CacheCalibrator （位于 torch_tensorrt.ts.ptq 中的类） CalibrationAlgo （位于 torch_tensorrt.ts.ptq 中的类） cdhw32 (torch_tensorrt.memory_format 属性) check_method_op_support() （位于模块 torch_tensorrt.ts 中） chw16 (torch_tensorrt.memory_format 属性) chw2 (torch_tensorrt.memory_format 属性) chw32 (torch_tensorrt.memory_format 属性) chw4 (torch_tensorrt.memory_format 属性) CompilationSettings （位于 torch_tensorrt.dynamo 中的类）, [1]

compile() （位于模块 torch_tensorrt 中） （位于模块 torch_tensorrt.dynamo 中） （位于模块 torch_tensorrt.fx 中） （位于模块 torch_tensorrt.ts 中） (torch_tensorrt.MutableTorchTensorRTModule 方法) convert_exported_program_to_serialized_trt_engine() （位于模块 torch_tensorrt.dynamo 中） convert_method_to_trt_engine() （位于模块 torch_tensorrt 中） （位于模块 torch_tensorrt.ts 中） cross_compile_for_windows() （位于模块 torch_tensorrt 中） （位于模块 torch_tensorrt.dynamo 中）

D

DataLoaderCalibrator （位于 torch_tensorrt.ts.ptq 中的类） debug （位于 torch_tensorrt.logging 中的类） Device （位于 torch_tensorrt 中的类） device_type (torch_tensorrt.Device 属性) DeviceType （位于 torch_tensorrt 中的类） dhwc (torch_tensorrt.memory_format 属性) dhwc8 (torch_tensorrt.memory_format 属性) disable_profiling() (torch_tensorrt.runtime.PythonTorchTensorRTModule 方法)

DLA (torch_tensorrt.DeviceType 属性) dla_core (torch_tensorrt.Device 属性) dla_hwc4 (torch_tensorrt.memory_format 属性) dla_linear (torch_tensorrt.memory_format 属性) DLA_STANDALONE (torch_tensorrt.EngineCapability 属性) 文档 dtype （位于 torch_tensorrt 中的类） (torch_tensorrt.Input 属性)

E

embed_engine_in_new_module() （位于模块 torch_tensorrt.ts 中） enable_cudagraphs() （位于模块 torch_tensorrt.runtime 中） enable_output_allocator() （位于模块 torch_tensorrt.runtime 中） enable_pre_allocated_outputs() （位于模块 torch_tensorrt.runtime 中） enable_profiling() (torch_tensorrt.runtime.PythonTorchTensorRTModule 方法)

EngineCapability （位于 torch_tensorrt 中的类） ENTROPY_CALIBRATION (torch_tensorrt.ts.ptq.CalibrationAlgo 属性) ENTROPY_CALIBRATION_2 (torch_tensorrt.ts.ptq.CalibrationAlgo 属性) errors （位于 torch_tensorrt.logging 中的类） example_tensor() (torch_tensorrt.Input 方法) export() （位于模块 torch_tensorrt.dynamo 中）

F

f16 (torch_tensorrt.dtype 属性) f32 (torch_tensorrt.dtype 属性) f64 (torch_tensorrt.dtype 属性) f8 (torch_tensorrt.dtype 属性)

format (torch_tensorrt.Input 属性) forward() (torch_tensorrt.runtime.PythonTorchTensorRTModule 方法) (torch_tensorrt.runtime.TorchTensorRTModule 方法) from_tensor() (torch_tensorrt.Input 类方法) from_tensors() (torch_tensorrt.Input 类方法)

G

get_cudagraphs_mode() （位于模块 torch_tensorrt.runtime 中） get_extra_state() (torch_tensorrt.runtime.TorchTensorRTModule 方法) get_layer_info() (torch_tensorrt.runtime.PythonTorchTensorRTModule 方法)

get_whole_cudagraphs_mode() （位于模块 torch_tensorrt.runtime 中） GPU (torch_tensorrt.DeviceType 属性) gpu_id (torch_tensorrt.Device 属性) graphs （位于 torch_tensorrt.logging 中的类）

H

hwc (torch_tensorrt.memory_format 属性)

hwc16 (torch_tensorrt.memory_format 属性) hwc8 (torch_tensorrt.memory_format 属性)

I

i32 (torch_tensorrt.dtype 属性) i64 (torch_tensorrt.dtype 属性) i8 (torch_tensorrt.dtype 属性)

info （位于 torch_tensorrt.logging 中的类） Input （位于 torch_tensorrt 中的类） InputTensorSpec （位于 torch_tensorrt.fx 中的类） internal_errors （位于 torch_tensorrt.logging 中的类）

L

LEGACY_CALIBRATION (torch_tensorrt.ts.ptq.CalibrationAlgo 属性) linear (torch_tensorrt.memory_format 属性)

load() （位于模块 torch_tensorrt 中） load_cross_compiled_exported_program() （位于模块 torch_tensorrt 中） （位于模块 torch_tensorrt.dynamo 中）

M

memory_format （位于 torch_tensorrt 中的类） MINMAX_CALIBRATION (torch_tensorrt.ts.ptq.CalibrationAlgo 属性) 模块 torch_tensorrt torch_tensorrt.dynamo, [1], [2], [3], [4] torch_tensorrt.fx torch_tensorrt.logging torch_tensorrt.runtime torch_tensorrt.ts torch_tensorrt.ts.ptq

MutableTorchTensorRTModule （位于 torch_tensorrt 中的类）

P

Python 增强提案 PEP 287

PythonTorchTensorRTModule （位于 torch_tensorrt.runtime 中的类）

R

阅读 refit_gm() (torch_tensorrt.MutableTorchTensorRTModule 方法)

refit_module_weights() （位于模块 torch_tensorrt.dynamo 中） RFC RFC 2822

S

SAFETY (torch_tensorrt.EngineCapability 属性) save() （位于模块 torch_tensorrt 中） save_cross_compiled_exported_program() （位于模块 torch_tensorrt.dynamo 中） set_cudagraphs_mode() （位于模块 torch_tensorrt.runtime 中）

set_expected_dynamic_shape_range() (torch_tensorrt.MutableTorchTensorRTModule 方法) set_extra_state() (torch_tensorrt.runtime.TorchTensorRTModule 方法) set_multi_device_safe_mode() （位于模块 torch_tensorrt.runtime 中） STANDARD (torch_tensorrt.EngineCapability 属性) STR （C 宏）

T

TensorRTCompileSpec() （位于模块 torch_tensorrt.ts 中） to() (torch_tensorrt.DeviceType 方法) (torch_tensorrt.dtype 方法) (torch_tensorrt.EngineCapability 方法) (torch_tensorrt.memory_format 方法) torch_tensorrt 模块 torch_tensorrt.dynamo module, [1], [2], [3], [4] torch_tensorrt.fx 模块 torch_tensorrt.logging 模块 torch_tensorrt.runtime 模块 torch_tensorrt.ts 模块 torch_tensorrt.ts.ptq 模块 torch_tensorrt::DataType （C++ 类） torch_tensorrt::DataType::DataType （C++ 函数）, [1], [2] torch_tensorrt::DataType::operator bool （C++ 函数） torch_tensorrt::DataType::operator Value （C++ 函数） torch_tensorrt::DataType::operator!= （C++ 函数）, [1] torch_tensorrt::DataType::operator== （C++ 函数）, [1] torch_tensorrt::DataType::Value （C++ 枚举） torch_tensorrt::DataType::Value::kBool （C++ 枚举成员） torch_tensorrt::DataType::Value::kChar （C++ 枚举成员） torch_tensorrt::DataType::Value::kDouble （C++ 枚举成员） torch_tensorrt::DataType::Value::kFloat （C++ 枚举成员） torch_tensorrt::DataType::Value::kHalf （C++ 枚举成员） torch_tensorrt::DataType::Value::kInt （C++ 枚举成员） torch_tensorrt::DataType::Value::kLong （C++ 枚举成员） torch_tensorrt::DataType::Value::kUnknown （C++ 枚举成员） torch_tensorrt::Device （C++ 结构体） torch_tensorrt::Device::allow_gpu_fallback （C++ 成员） torch_tensorrt::Device::Device （C++ 函数） torch_tensorrt::Device::device_type （C++ 成员） torch_tensorrt::Device::DeviceType （C++ 类）, [1] torch_tensorrt::Device::DeviceType::DeviceType （C++ 函数）, [1], [2], [3], [4], [5] torch_tensorrt::Device::DeviceType::operator bool （C++ 函数）, [1] torch_tensorrt::Device::DeviceType::operator Value （C++ 函数）, [1] torch_tensorrt::Device::DeviceType::operator!= （C++ 函数）, [1] torch_tensorrt::Device::DeviceType::operator== （C++ 函数）, [1] torch_tensorrt::Device::DeviceType::Value （C++ 枚举）, [1] torch_tensorrt::Device::DeviceType::Value::kDLA （C++ 枚举成员）, [1] torch_tensorrt::Device::DeviceType::Value::kGPU （C++ 枚举成员）, [1] torch_tensorrt::Device::dla_core （C++ 成员） torch_tensorrt::Device::gpu_id （C++ 成员） torch_tensorrt::dump_build_info （C++ 函数） torch_tensorrt::EngineCapability （C++ 枚举） torch_tensorrt::EngineCapability::kDLA_STANDALONE （C++ 枚举成员） torch_tensorrt::EngineCapability::kSAFETY （C++ 枚举成员） torch_tensorrt::EngineCapability::kSTANDARD （C++ 枚举成员） torch_tensorrt::get_build_info （C++ 函数） torch_tensorrt::GraphInputs （C++ 结构体） torch_tensorrt::GraphInputs::input_signature （C++ 成员） torch_tensorrt::GraphInputs::inputs （C++ 成员） torch_tensorrt::Input （C++ 结构体） torch_tensorrt::Input::dtype （C++ 成员） torch_tensorrt::Input::format （C++ 成员） torch_tensorrt::Input::Input （C++ 函数）, [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14], [15], [16], [17] torch_tensorrt::Input::max_shape （C++ 成员） torch_tensorrt::Input::min_shape （C++ 成员） torch_tensorrt::Input::opt_shape （C++ 成员） torch_tensorrt::Input::shape （C++ 成员） torch_tensorrt::Input::tensor_domain （C++ 成员） torch_tensorrt::logging::get_is_colored_output_on （C++ 函数） torch_tensorrt::logging::get_logging_prefix （C++ 函数） torch_tensorrt::logging::get_reportable_log_level （C++ 函数） torch_tensorrt::logging::Level （C++ 枚举） torch_tensorrt::logging::Level::kDEBUG （C++ 枚举成员） torch_tensorrt::logging::Level::kERROR （C++ 枚举成员） torch_tensorrt::logging::Level::kGRAPH （C++ 枚举成员） torch_tensorrt::logging::Level::kINFO （C++ 枚举成员） torch_tensorrt::logging::Level::kINTERNAL_ERROR （C++ 枚举成员）

torch_tensorrt::logging::Level::kWARNING （C++ 枚举成员） torch_tensorrt::logging::log （C++ 函数） torch_tensorrt::logging::set_is_colored_output_on （C++ 函数） torch_tensorrt::logging::set_logging_prefix （C++ 函数） torch_tensorrt::logging::set_reportable_log_level （C++ 函数） torch_tensorrt::ptq::Int8CacheCalibrator （C++ 类） torch_tensorrt::ptq::Int8CacheCalibrator::getBatch （C++ 函数） torch_tensorrt::ptq::Int8CacheCalibrator::getBatchSize （C++ 函数） torch_tensorrt::ptq::Int8CacheCalibrator::Int8CacheCalibrator （C++ 函数） torch_tensorrt::ptq::Int8CacheCalibrator::operator nvinfer1::IInt8Calibrator* （C++ 函数） torch_tensorrt::ptq::Int8CacheCalibrator::readCalibrationCache （C++ 函数） torch_tensorrt::ptq::Int8CacheCalibrator::writeCalibrationCache （C++ 函数） torch_tensorrt::ptq::Int8Calibrator （C++ 类） torch_tensorrt::ptq::Int8Calibrator::getBatch （C++ 函数） torch_tensorrt::ptq::Int8Calibrator::getBatchSize （C++ 函数） torch_tensorrt::ptq::Int8Calibrator::Int8Calibrator （C++ 函数） torch_tensorrt::ptq::Int8Calibrator::operator nvinfer1::IInt8Calibrator* （C++ 函数） torch_tensorrt::ptq::Int8Calibrator::readCalibrationCache （C++ 函数） torch_tensorrt::ptq::Int8Calibrator::writeCalibrationCache （C++ 函数） torch_tensorrt::set_device （C++ 函数） torch_tensorrt::TensorFormat （C++ 类） torch_tensorrt::TensorFormat::operator bool （C++ 函数） torch_tensorrt::TensorFormat::operator Value （C++ 函数） torch_tensorrt::TensorFormat::operator!= （C++ 函数）, [1] torch_tensorrt::TensorFormat::operator== （C++ 函数）, [1] torch_tensorrt::TensorFormat::TensorFormat （C++ 函数）, [1], [2] torch_tensorrt::TensorFormat::Value （C++ 枚举） torch_tensorrt::TensorFormat::Value::kChannelsLast （C++ 枚举成员） torch_tensorrt::TensorFormat::Value::kContiguous （C++ 枚举成员） torch_tensorrt::TensorFormat::Value::kUnknown （C++ 枚举成员） torch_tensorrt::torchscript::check_method_operator_support （C++ 函数） torch_tensorrt::torchscript::compile （C++ 函数） torch_tensorrt::torchscript::CompileSpec （C++ 结构体） torch_tensorrt::torchscript::CompileSpec::allow_shape_tensors （C++ 成员） torch_tensorrt::torchscript::CompileSpec::capability （C++ 成员） torch_tensorrt::torchscript::CompileSpec::CompileSpec （C++ 函数）, [1], [2], [3] torch_tensorrt::torchscript::CompileSpec::debug （C++ 成员） torch_tensorrt::torchscript::CompileSpec::device （C++ 成员） torch_tensorrt::torchscript::CompileSpec::disable_tf32 （C++ 成员） torch_tensorrt::torchscript::CompileSpec::dla_global_dram_size （C++ 成员） torch_tensorrt::torchscript::CompileSpec::dla_local_dram_size （C++ 成员） torch_tensorrt::torchscript::CompileSpec::dla_sram_size （C++ 成员） torch_tensorrt::torchscript::CompileSpec::enabled_precisions （C++ 成员） torch_tensorrt::torchscript::CompileSpec::graph_inputs （C++ 成员） torch_tensorrt::torchscript::CompileSpec::min_block_size （C++ 成员） torch_tensorrt::torchscript::CompileSpec::num_avg_timing_iters （C++ 成员） torch_tensorrt::torchscript::CompileSpec::ptq_calibrator （C++ 成员） torch_tensorrt::torchscript::CompileSpec::refit （C++ 成员） torch_tensorrt::torchscript::CompileSpec::require_full_compilation （C++ 成员） torch_tensorrt::torchscript::CompileSpec::sparse_weights （C++ 成员） torch_tensorrt::torchscript::CompileSpec::torch_executed_modules （C++ 成员） torch_tensorrt::torchscript::CompileSpec::torch_executed_ops （C++ 成员） torch_tensorrt::torchscript::CompileSpec::truncate_long_and_double （C++ 成员） torch_tensorrt::torchscript::CompileSpec::workspace_size （C++ 成员） torch_tensorrt::torchscript::convert_method_to_trt_engine （C++ 函数） torch_tensorrt::torchscript::embed_engine_in_new_module （C++ 函数） TORCH_TENSORRT_MAJOR_VERSION （C 宏） TORCH_TENSORRT_MINOR_VERSION （C 宏） TORCH_TENSORRT_PATCH_VERSION （C 宏） TORCH_TENSORRT_PTQ_DEPRECATION （C 宏） TORCH_TENSORRT_VERSION （C 宏） TorchTensorRTModule （位于 torch_tensorrt.runtime 中的类） TORCHTRT_API （C 宏） TORCHTRT_HIDDEN （C 宏） trace() （位于模块 torch_tensorrt.dynamo 中） TRTInterpreter （位于 torch_tensorrt.fx 中的类） TRTInterpreterResult （位于 torch_tensorrt.fx 中的类） TRTModule （位于 torch_tensorrt.fx 中的类） try_from() (torch_tensorrt.DeviceType 类方法) (torch_tensorrt.dtype 类方法) (torch_tensorrt.EngineCapability 类方法) (torch_tensorrt.memory_format 类方法) try_to() (torch_tensorrt.DeviceType 方法) (torch_tensorrt.dtype 方法) (torch_tensorrt.EngineCapability 方法) (torch_tensorrt.memory_format 方法)

U

u8 (torch_tensorrt.dtype 属性)

UNKNOWN (torch_tensorrt.DeviceType 属性) unknown (torch_tensorrt.dtype 属性)

V

validate_input_shapes() (torch_tensorrt.runtime.PythonTorchTensorRTModule 方法)

W

warnings （位于 torch_tensorrt.logging 中的类）

weight_streaming() （位于模块 torch_tensorrt.runtime 中） 编写

X

XSTR （C 宏）