text: # Owner(s): ["oncall: aiacc"]
  
  import torch
  import torch.fx.experimental.fx_acc.acc_ops as acc_ops
  import torch.nn as nn
  from torch.testing._internal.common_fx2trt import AccTestCase, InputTensorSpec
  from torch.testing._internal.common_utils import run_tests
  
  
  class TestSoftsignConverter(AccTestCase):
      def test_softsign(self):
          class TestModule(nn.Module):
              def forward(self, x):
                  return nn.functional.softsign(x)
  
          inputs = [torch.randn(1, 10)]
          self.run_test(TestModule(), inputs, expected_ops={acc_ops.softsign})
  
      def test_softsign_with_dynamic_shape(self):
          class TestModule(nn.Module):
              def forward(self, x):
                  return nn.functional.softsign(x)
  
          input_specs = [
              InputTensorSpec(
                  shape=(-1, -1, -1),
                  dtype=torch.float32,
                  shape_ranges=[((1, 1, 1), (1, 2, 3), (3, 3, 3))],
              ),
          ]
          self.run_test_with_dynamic_shape(
              TestModule(), input_specs, expected_ops={acc_ops.softsign}
          )
  
  if __name__ == '__main__':
      run_tests()
  