Skip to content

# File torch_tensorrt.h

↰ Parent directory ( `cpp/include/torch_tensorrt` )

Contents

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

- Includes

- Namespaces

- Classes

- Enums

- Functions

## 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