Function trtorch::CompileGraph ¶
-
Defined in File trtorch.h
Function Documentation ¶
-
TRTORCH_API torch::jit::Module
trtorch
::
CompileGraph
( const torch::jit::Module & module , CompileSpec info )
-
Compile a TorchScript module for NVIDIA GPUs using TensorRT.
Takes a existing TorchScript module and a set of settings to configure the compiler and will convert methods to JIT Graphs which call equivalent TensorRT engines
- Parameters
-
-
module
: torch::jit::Module - Existing TorchScript module -
info
: trtorch::CompileSpec - Compilation settings
-
Converts specifically the forward method of a TorchScript Module
- Return
-
: A new module trageting a TensorRT engine