File trtorch.h

↰ Parent directory ( cpp/include/trtorch )

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

• Program Listing for File trtorch.h

Includes

• cuda_runtime.h

• iostream

• memory

• set

• string

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

• vector

Namespaces

• Namespace trtorch

Classes

• Struct CompileSpec

• Struct CompileSpec::Device

• Struct CompileSpec::Input

• Struct CompileSpec::InputRange

• Struct CompileSpec::TorchFallback

• Class CompileSpec::DataType

• Class Device::DeviceType

• Class CompileSpec::TensorFormat

Functions

• Function trtorch::CheckMethodOperatorSupport

• Function trtorch::CompileGraph

• Function trtorch::ConvertGraphToTRTEngine

• Function trtorch::dump_build_info

• Function trtorch::EmbedEngineInNewModule

• Function trtorch::get_build_info

• Function trtorch::set_device