Searched refs:initial_sum (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 790 Value initial_sum = rewriter.create<mhlo::PowOp>(loc, q, neg_x); in MaterializeZeta() local
795 initial_sum = rewriter.create<mhlo::AddOp>(loc, initial_sum, neg_power); in MaterializeZeta()
804 Value s = rewriter.create<mhlo::AddOp>(loc, initial_sum, in MaterializeZeta()
852 Value abs_initial_sum = rewriter.create<mhlo::AbsOp>(loc, initial_sum); in MaterializeZeta()
861 initial_sum, s); in MaterializeZeta()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 1923 XlaOp initial_sum = Pow(q, Neg(x)); in Zeta() local
1927 initial_sum = initial_sum + neg_power; in Zeta()
1931 XlaOp s = initial_sum + neg_power * a / (x - ScalarLike(a, 1.)); in Zeta()
1954 Select(Lt(Abs(neg_power), Abs(initial_sum) * Epsilon(&builder, type)), in Zeta()
1955 initial_sum, s); in Zeta()
|