change function calls from nn.Module to nn.functional

Author: zewenli98

py/torch_tensorrt/dynamo/conversion/impl/activation/ops.py

@@ -137,8 +137,8 @@ def elu(
 
         def elu_dyn_range_fn(dyn_range):
             return (
-                torch.nn.ELU(dyn_range[0], alpha),
-                torch.nn.ELU(dyn_range[1], alpha),
+                torch.nn.functional.elu(dyn_range[0], alpha),
+                torch.nn.functional.elu(dyn_range[1], alpha),
             )
 
         return convert_activation(
@@ -163,7 +163,10 @@ def selu(
     operation_type = trt.ActivationType.SELU
 
     def selu_dyn_range_fn(dyn_range):
-        return (torch.nn.SELU(dyn_range[0]), torch.nn.SELU(dyn_range[1]))
+        return (
+            torch.nn.functional.selu(dyn_range[0]),
+            torch.nn.functional.selu(dyn_range[1]),
+        )
 
     return convert_activation(
         network,
@@ -187,7 +190,10 @@ def softsign(
     operation_type = trt.ActivationType.SOFTSIGN
 
     def softsign_dyn_range_fn(dyn_range):
-        return (torch.nn.Softsign(dyn_range[0]), torch.nn.Softsign(dyn_range[1]))
+        return (
+            torch.nn.functional.softsign(dyn_range[0]),
+            torch.nn.functional.softsign(dyn_range[1]),
+        )
 
     return convert_activation(
         network,
@@ -212,8 +218,8 @@ def softplus(
 
     def softplus_dyn_range_fn(dyn_range):
         return (
-            torch.nn.Softplus(dyn_range[0], beta),
-            torch.nn.Softplus(dyn_range[1], beta),
+            torch.nn.functional.softplus(dyn_range[0], beta),
+            torch.nn.functional.softplus(dyn_range[1], beta),
         )
 
     return convert_activation(
@@ -303,7 +309,7 @@ def scaled_tanh(
 
     def scaled_tanh_dyn_range_fn(dyn_range):
         def scaled_tanh_fn(x):
-            return alpha * torch.nn.Tanh(beta * x)
+            return alpha * torch.nn.functional.tanh(beta * x)
 
         return scaled_tanh_fn(dyn_range[0]), scaled_tanh_fn(dyn_range[1])