You can see the call where the engine is executed, based on a constant which is the ID of the engine, telling JIT how to find the engine and the input tensor which will be fed to TensorRT.The engine represents the exact same calculations as what is done by running a normal PyTorch module but optimized to run on your GPU.