We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 819c911 commit 78f808bCopy full SHA for 78f808b
cpp/include/torch_tensorrt/torch_tensorrt.h
@@ -739,7 +739,12 @@ TORCHTRT_API std::string convert_method_to_trt_engine(
739
* module. Registers execution of the engine as the forward method of the module
740
* Forward is defined as: forward(Tensor[]) -> Tensor[]
741
*
742
- * @return: A new module trageting a TensorRT engine
+ * TensorRT bindings must have names with the following format:
743
+ * - [symbol].[index in input / output array]
744
+ * ex.
745
+ * - [x.0, x.1, x.2] -> [y.0]
746
+ *
747
+ * @return: A new module targeting a TensorRT engine
748
*/
749
TORCHTRT_API torch::jit::Module embed_engine_in_new_module(const std::string& engine, Device device);
750
} // namespace torchscript
0 commit comments