Skip to content

# Function trtorch::EmbedEngineInNewModule

- Defined in File trtorch.h

## Function Documentation

TRTORCH_API torch::jit::Module `trtorch `::`` `EmbedEngineInNewModule` ( const std::string & engine )

  Take a previously created TensorRT engine and embed it in in a TorchScript module.

  Takes a pre-built serialized TensorRT engine and embeds it in a TorchScript module. Registers execution of the engine as the forward method of the module Forward is defined as: forward(Tensor[]) -> Tensor[]

  Parameters

    - `engine` : std::string - Pre-built serialized TensorRT engine

  Return

    : A new module trageting a TensorRT engine