Searched refs:neg_power (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 788 Value neg_power = zero_like_a; in MaterializeZeta() local
794 neg_power = rewriter.create<mhlo::PowOp>(loc, a, neg_x); in MaterializeZeta()
795 initial_sum = rewriter.create<mhlo::AddOp>(loc, initial_sum, neg_power); in MaterializeZeta()
798 neg_power = rewriter.create<mhlo::PowOp>(loc, a, neg_x); in MaterializeZeta()
801 Value neg_power_mul_a = rewriter.create<mhlo::MulOp>(loc, neg_power, a); in MaterializeZeta()
830 chlo::getConstantLike(rewriter, loc, .5, neg_power); in MaterializeZeta()
835 loc, neg_power, in MaterializeZeta()
851 Value abs_neg_power = rewriter.create<mhlo::AbsOp>(loc, neg_power); in MaterializeZeta()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 1922 XlaOp neg_power = ScalarLike(a, 0.); in Zeta() local
1926 neg_power = Pow(a, Neg(x)); in Zeta()
1927 initial_sum = initial_sum + neg_power; in Zeta()
1930 neg_power = Pow(a, Neg(x)); in Zeta()
1931 XlaOp s = initial_sum + neg_power * a / (x - ScalarLike(a, 1.)); in Zeta()
1945 s = s + neg_power * in Zeta()
1946 (ScalarLike(neg_power, 0.5) + in Zeta()
1954 Select(Lt(Abs(neg_power), Abs(initial_sum) * Epsilon(&builder, type)), in Zeta()
|