diff --git a/experimental/torch_xla2/torch_xla2/ops/jaten.py b/experimental/torch_xla2/torch_xla2/ops/jaten.py index 08f27de7940..6160a61fb6d 100644 --- 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,