Searched refs:FullLike (Results 1 – 4 of 4) sorted by relevance
| /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/client/lib/ |
| D | math.cc | 268 return Select(lt, FullLike(x, w_less_than_5_constants[i]), in ErfInv()
269 FullLike(x, w_greater_than_5_constants[i])); in ErfInv()
404 XlaOp inf_bcast = FullLike(input, std::numeric_limits<float>::infinity()); in Lgamma()
478 FullLike(input, std::numeric_limits<float>::quiet_NaN()), in Digamma()
517 return Select(Ne(x, FullLike(x, -1)), in Acos()
520 FullLike(x, M_PI)); in Acos()
|
| D | constants.h | 99 XlaOp FullLike(XlaOp prototype, T value) { in FullLike() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 145 auto ones = FullLike(diags, 1); in InvertDiagonalBlocks()
|