)>;

//===----------------------------------------------------------------------===//
// Sqrt op patterns.
//===----------------------------------------------------------------------===//

// SqrtGrad(y, dy) = dy * 0.5 / y
def LowerSqrtGradOp : Pat<
  (TF_SqrtGradOp $y, $dy),
  (TF_DivOp
    (TF_MulOp $dy, (TF_ConstOp (GetScalarOfFloatType<"0.5"> $dy))),
    $y
  )>;

    static bool isCompatibleReturnTypes(TypeRange inferred, TypeRange actual) {
      return ArraysAreCastCompatible(inferred, actual);
    }
  }];
}

def TF_SqrtGradOp : TF_Op<"SqrtGrad", [Pure, TF_SameOperandsAndResultTypeResolveRef]> {
  let summary = "Computes the gradient for the sqrt of `x` wrt its input.";

  let description = [{