From d3a1dd938bb3c523d061fb95092690abba8bcb59 Mon Sep 17 00:00:00 2001 From: John Welsh Date: Tue, 26 Jul 2022 12:59:33 -0700 Subject: [PATCH] added add inplace converter --- torch2trt/converters/add.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch2trt/converters/add.py b/torch2trt/converters/add.py index 6bdf2e63..c80250be 100644 --- a/torch2trt/converters/add.py +++ b/torch2trt/converters/add.py @@ -2,6 +2,7 @@ from torch2trt.module_test import add_module_test +@tensorrt_converter('torch.Tensor.add_') @tensorrt_converter('torch.add') @tensorrt_converter('torch.Tensor.__iadd__') @tensorrt_converter('torch.Tensor.__add__')