Torch-TensorRT C++ API#
Class Hierarchy#
-
- Namespace torch_tensorrt
- Namespace torch_tensorrt::logging
- Enum Level
- Namespace torch_tensorrt::torchscript
- Struct CompileSpec
- Struct Device
- Class Device::DeviceType
- Struct GraphInputs
- Struct Input
- Class DataType
- Class TensorFormat
- Enum EngineCapability
- Namespace torch_tensorrt::logging
- Namespace torch_tensorrt
File Hierarchy#
-
- Directory cpp
- Directory include
- Directory torch_tensorrt
- File logging.h
- File macros.h
- File torch_tensorrt.h
- Directory torch_tensorrt
- Directory include
- Directory cpp
Full API#
Namespaces#
Classes and Structs#
- Struct CompileSpec
- Struct Documentation
torch_tensorrt::torchscript::CompileSpecCompileSpec()CompileSpec()CompileSpec()CompileSpec()graph_inputsenabled_precisionsdisable_tf32sparse_weightsrefitdebugtruncate_long_and_doubleallow_shape_tensorsdevicecapabilitynum_avg_timing_itersworkspace_sizedla_sram_sizedla_local_dram_sizedla_global_dram_sizerequire_full_compilationmin_block_sizetorch_executed_opstorch_executed_modules
- Struct Documentation