File torch_tensorrt.h

↰ Parent directory ( cpp/include/torch_tensorrt )

Definition ( cpp/include/torch_tensorrt/torch_tensorrt.h )

• Program Listing for File torch_tensorrt.h

Includes

• cuda_runtime.h

• iostream

• memory

• set

• string

• torch_tensorrt/macros.h ( File macros.h )

• vector

Namespaces

• Namespace torch_tensorrt

• Namespace torch_tensorrt::torchscript

Classes

• Struct Device

• Struct Input

• Struct CompileSpec

• Class DataType

• Class Device::DeviceType

• Class TensorFormat

Enums

• Enum EngineCapability

Functions

• Function torch_tensorrt::dump_build_info

• Function torch_tensorrt::get_build_info

• Function torch_tensorrt::set_device

• Function torch_tensorrt::torchscript::check_method_operator_support

• Function torch_tensorrt::torchscript::compile

• Function torch_tensorrt::torchscript::convert_method_to_trt_engine

• Function torch_tensorrt::torchscript::embed_engine_in_new_module