Searched refs:FullLike (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 370 return Select(lt, FullLike(x, w_less_than_5_constants[i]), in ErfInv32()
371 FullLike(x, w_greater_than_5_constants[i])); in ErfInv32()
437 auto c = FullLike(x, w_less_than_6_25_constants[i]); in ErfInv64()
439 c = Select(lt_6_25, c, FullLike(x, w_less_than_16_constants[i])); in ErfInv64()
442 c = Select(lt_16, c, FullLike(x, w_greater_than_16_constants[i])); in ErfInv64()
601 XlaOp inf_bcast = FullLike(input, std::numeric_limits<float>::infinity()); in Lgamma()
699 FullLike(input, std::numeric_limits<float>::quiet_NaN()), in Digamma()
768 enabled, a, FullLike(a, 1), FullLike(a, 1), x, FullLike(a, 0), in IgammaSeries()
769 FullLike(a, 0), in IgammaSeries()
830 t = Select(qk_is_nonzero, Abs((ans - r) / r), FullLike(t, 1)); in IgammacContinuedFraction()
[all …]
|
| D | constants.h | 105 XlaOp FullLike(XlaOp prototype, T value) { in FullLike() function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | constants_test.cc | 184 TEST_F(ConstantsTest, FullLike) { in TEST_F() argument
187 auto val2 = FullLike(val1, 10); in TEST_F()
195 FullLike(tuple, 10); // Illegal; can't do FullLike on a tuple. in TEST_F()
202 auto scalar2 = FullLike(scalar1, 2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 214 FullLike(l, std::numeric_limits<float>::quiet_NaN()), l); in BuildCholesky()
|
| D | triangular_solve_expander.cc | 265 auto ones = FullLike(diags, 1); in InvertDiagonalBlocks()
|