diff --git a/CONTRIBUTORS.md b/CONTRIBUTORS.md index cd3a32f5..d87d8d0c 100644 --- a/CONTRIBUTORS.md +++ b/CONTRIBUTORS.md @@ -5,6 +5,7 @@ who have agreed to torch2trt's Contributor License Agreement. - [John Welsh](https://github.com/jaybdub) (CLA) - John Welsh +- [Eunseop Shin](https://github.com/kairos03) (CLA) ## Becoming a Contributor diff --git a/torch2trt/converters/transpose.py b/torch2trt/converters/transpose.py index c5131592..440cf274 100644 --- a/torch2trt/converters/transpose.py +++ b/torch2trt/converters/transpose.py @@ -3,6 +3,7 @@ @tensorrt_converter("torch.transpose", enabled=trt_version() < '7.0') +@tensorrt_converter("torch.Tensor.transpose", enabled=trt_version() < '7.0') def convert_transpose(ctx): input = ctx.method_args[0] input_trt = add_missing_trt_tensors(ctx.network, [input])[0] @@ -19,6 +20,7 @@ def convert_transpose(ctx): @tensorrt_converter('torch.transpose', enabled=trt_version() >= '7.0') +@tensorrt_converter('torch.Tensor.transpose', enabled=trt_version() >= '7.0') def convert_transpose_trt7(ctx): input = ctx.method_args[0] input_trt = add_missing_trt_tensors(ctx.network, [input])[0]