Delete unused tracing functions from ops.cpp.

- `SgnOp` and `SignOp`
    - Full codegen migration: #3577
    - Mistakenly re-introduced: #3572
- `LogSigmoid`
    - Introduced: #3539
    - Full codegen migration: #3743
- `SiLU`
    - Introduced: #2721
    - Full codegen migration: #3780
- `SiLUBackward`
    - Introduced: #3195
    - Full codegen migration: #3780
- `SeLU`
    - Introduced: #3547
    - Full codegen migration: #3780
- `Sigmoid`
    - Introduced: 6a73deb (no PR record)
    - Full codegen migration: #6342

Author: ysiraichi

torch_xla/csrc/ops/ops.cpp

@@ -116,26 +116,6 @@ torch::lazy::NodePtr Logit(const torch::lazy::Value& input,
                    torch::lazy::MHash(eps));
 }
 
-torch::lazy::NodePtr SgnOp(const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(0));
-    return node.ReturnOp(BuildSgn(xla_input), loctx);
-  };
-  return GenericOp(torch::lazy::OpKind(at::aten::sgn), {input},
-                   GetXlaShape(input), std::move(lower_fn));
-}
-
-torch::lazy::NodePtr SignOp(const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(0));
-    return node.ReturnOp(BuildSign(xla_input), loctx);
-  };
-  return GenericOp(torch::lazy::OpKind(at::aten::sign), {input},
-                   GetXlaShape(input), std::move(lower_fn));
-}
-
 torch::lazy::NodePtr Prelu(const torch::lazy::Value& input,
                            const torch::lazy::Value& weight) {
   auto lower_fn = [](const XlaNode& node,
@@ -169,57 +149,6 @@ torch::lazy::NodePtr PreluBackward(const torch::lazy::Value& grad,
       std::move(lower_fn), /*num_outputs=*/2);
 }
 
-torch::lazy::NodePtr LogSigmoid(const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(0));
-    return node.ReturnOps(BuildLogSigmoid(xla_input), loctx);
-  };
-  return GenericOp(torch::lazy::OpKind(at::aten::log_sigmoid), {input},
-                   GetXlaShape(input), std::move(lower_fn), /*num_outputs=*/2);
-}
-
-torch::lazy::NodePtr SiLU(const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(0));
-    return node.ReturnOp(xla_input * BuildSigmoid(xla_input), loctx);
-  };
-  return GenericOp(torch::lazy::OpKind(at::aten::silu), {input},
-                   GetXlaShape(input), std::move(lower_fn));
-}
-
-torch::lazy::NodePtr SiLUBackward(const torch::lazy::Value& grad_output,
-                                  const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_grad_output = loctx->GetOutputOp(node.operand(0));
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(1));
-    return node.ReturnOp(BuildSiLUBackward(xla_grad_output, xla_input), loctx);
-  };
-  auto lower_for_shape_fn =
-      [](absl::Span<const xla::XlaOp> operands) -> xla::XlaOp {
-    return BuildSiLUBackward(operands[0], operands[1]);
-  };
-  return GenericOp(
-      torch::lazy::OpKind(at::aten::silu_backward), {grad_output, input},
-      [&]() {
-        return InferOutputShape({GetXlaShape(grad_output), GetXlaShape(input)},
-                                lower_for_shape_fn);
-      },
-      std::move(lower_fn));
-}
-
-torch::lazy::NodePtr Sigmoid(const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(0));
-    return node.ReturnOp(BuildSigmoid(xla_input), loctx);
-  };
-  return GenericOp(torch::lazy::OpKind(at::aten::sigmoid), {input},
-                   GetXlaShape(input), std::move(lower_fn));
-}
-
 torch::lazy::NodePtr SigmoidBackward(const torch::lazy::Value& grad_output,
                                      const torch::lazy::Value& output) {
   torch::lazy::Value scalar_1 = ScalarOp(1, GetXlaShape(output));
@@ -958,16 +887,6 @@ torch::lazy::NodePtr Softplus(const torch::lazy::Value& input,
                    std::move(lower_fn));
 }
 
-torch::lazy::NodePtr Selu(const torch::lazy::Value& input) {
-  auto lower_fn = [](const XlaNode& node,
-                     LoweringContext* loctx) -> XlaOpVector {
-    xla::XlaOp xla_input = loctx->GetOutputOp(node.operand(0));
-    return node.ReturnOp(BuildSelu(xla_input), loctx);
-  };
-  return GenericOp(torch::lazy::OpKind(at::aten::selu), {input},
-                   GetXlaShape(input), std::move(lower_fn));
-}
-
 torch::lazy::NodePtr ViewAsComplexCopy(const torch::lazy::Value& input) {
   auto lower_fn = [](const XlaNode& node,
                      LoweringContext* loctx) -> XlaOpVector {

torch_xla/csrc/ops/ops.h

@@ -80,10 +80,6 @@ torch::lazy::NodePtr Tan(const torch::lazy::Value& input);
 
 torch::lazy::NodePtr Neg(const torch::lazy::Value& input);
 
-torch::lazy::NodePtr SgnOp(const torch::lazy::Value& input);
-
-torch::lazy::NodePtr SignOp(const torch::lazy::Value& input);
-
 torch::lazy::NodePtr Min(const torch::lazy::Value& input,
                          const torch::lazy::Value& other);
 
@@ -114,15 +110,6 @@ torch::lazy::NodePtr Pow(const torch::lazy::Value& input,
 torch::lazy::NodePtr Fmod(const torch::lazy::Value& dividend,
                           const torch::lazy::Value& divisor);
 
-torch::lazy::NodePtr LogSigmoid(const torch::lazy::Value& input);
-
-torch::lazy::NodePtr Sigmoid(const torch::lazy::Value& input);
-
-torch::lazy::NodePtr SiLU(const torch::lazy::Value& input);
-
-torch::lazy::NodePtr SiLUBackward(const torch::lazy::Value& grad_output,
-                                  const torch::lazy::Value& input);
-
 torch::lazy::NodePtr SigmoidBackward(const torch::lazy::Value& grad_output,
                                      const torch::lazy::Value& output);
 
@@ -245,8 +232,6 @@ torch::lazy::NodePtr Softplus(const torch::lazy::Value& input,
                               const torch::lazy::Value& beta,
                               const torch::lazy::Value& threshold);
 
-torch::lazy::NodePtr Selu(const torch::lazy::Value& input);
-
 torch::lazy::NodePtr ViewAsComplexCopy(const torch::lazy::Value& input);
 
 torch::lazy::NodePtr ViewAsRealCopy(const torch::lazy::Value& input);