Add inplace ops for sigmoid and tanh (#8358)

pytorch · Nov 6, 2024 · 77abab2 · 77abab2
1 parent 7d6a4f2
commit 77abab2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/experimental/torch_xla2/torch_xla2/ops/jaten.py b/experimental/torch_xla2/torch_xla2/ops/jaten.py
@@ -45,6 +45,8 @@
   torch.ops.aten.squeeze_: (torch.ops.aten.squeeze, True),
   torch.ops.aten.clamp_: torch.ops.aten.clamp,
   torch.ops.aten.clamp_min_: torch.ops.aten.clamp_min,
+  torch.ops.aten.sigmoid_: torch.ops.aten.sigmoid,
+  torch.ops.aten.tanh_: torch.ops.aten.tanh,
   torch.ops.aten.ceil_: torch.ops.aten.ceil,
   torch.ops.aten.logical_not_: torch.ops.aten.logical_not,
   torch.ops.aten.unsqueeze_: torch.ops.aten.unsqueeze,