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__')