Searched refs:kLanczosCoefficients (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 498 constexpr std::array<double, 8> kLanczosCoefficients = { variable
535 for (int i = 0, end = kLanczosCoefficients.size(); i < end; ++i) { in MaterializeLgamma()
536 Value coeff = getConstantLike(rewriter, loc, kLanczosCoefficients[i], x); in MaterializeLgamma()
683 for (int i = 0, end = kLanczosCoefficients.size(); i < end; ++i) { in MaterializeDigamma()
684 Value coeff = getConstantLike(rewriter, loc, kLanczosCoefficients[i], x); in MaterializeDigamma()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 497 static constexpr std::array<double, 8> kLanczosCoefficients = { variable
532 for (int i = 0, end = kLanczosCoefficients.size(); i < end; ++i) { in Lgamma()
533 XlaOp lanczos_coefficient = ScalarLike(input, kLanczosCoefficients[i]); in Lgamma()
668 for (int i = 0, end = kLanczosCoefficients.size(); i < end; ++i) { in Digamma()
669 XlaOp lanczos_coefficient = ScalarLike(input, kLanczosCoefficients[i]); in Digamma()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 514 static constexpr std::array<double, 8> kLanczosCoefficients = { variable
602 for (int i = 0, end = kLanczosCoefficients.size(); i < end; ++i) { in matchAndRewrite()
603 Value lanczos_coefficient = create_const_op(kLanczosCoefficients[i]); in matchAndRewrite()
|