register torch.Tensor.softmax converter
@@ -2,6 +2,7 @@ from torch2trt.torch2trt import *
from torch2trt.module_test import add_module_test
+@tensorrt_converter('torch.Tensor.softmax')
@tensorrt_converter('torch.nn.functional.softmax')
def convert_softmax(ctx):
input = ctx.method_args[0]
