message: Rolling back MlirXlaOpKernel implementation of TanhGrad, since MLIR legalization does not handle binary ops with dynamic bounded shapes.
diff: @@ -227,7 +227,9 @@ XLA_MAKE_BINARY(SoftsignGrad,
  xla::Square(xla::Add(XlaHelpers::One(b, input_type(0)),
  xla::Abs(rhs)))));
  -REGISTER_XLA_OP(Name("TanhGrad"), MlirXlaOpKernel);
  +XLA_MAKE_BINARY(TanhGrad,
  + xla::Mul(rhs, xla::Sub(XlaHelpers::One(b, input_type(0)),
  + xla::Mul(lhs, lhs))));
  XLA_MAKE_BINARY(Pow, xla::Pow(lhs, rhs, extend_dimensions));
  