Searched refs:ConstantR0WithType (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | constants_test.cc | 33 ConstantR0WithType(&builder, xla::S32, 4); in XLA_TEST_F()
39 ConstantR0WithType(&builder, xla::S32, 4.5); in XLA_TEST_F()
47 ConstantR0WithType(&builder, xla::F32, -7); in XLA_TEST_F()
49 ConstantR0WithType(&builder, xla::F32, 0.5); in XLA_TEST_F()
|
| D | prng.cc | 131 auto const32 = ConstantR0WithType(builder, U64, 32); in Uint64ToUint32s()
141 ConstantR0WithType(builder, U64, 32)); in Uint32sToUint64()
180 ShiftRightLogical(dist, ConstantR0WithType(builder, unsigned_type, 1)); in StatelessRngUniformInt()
|
| D | constants.h | 34 XlaOp ConstantR0WithType(XlaBuilder* builder, PrimitiveType type, T value) { in ConstantR0WithType() function
89 return ConstantR0WithType(builder, shape.element_type(), value); in ScalarLike()
|
| D | svd_test.cc | 120 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
122 ConstantR0WithType(builder, F32, size); in GetAverageAbsoluteError()
|
| D | self_adjoint_eig_test.cc | 121 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
123 ConstantR0WithType(builder, F32, size); in GetAverageAbsoluteError()
|
| D | math.cc | 116 ConstantR0WithType(&b, U64, uint64{1} << 63)); in IsNegZero()
119 ConstantR0WithType(&b, U32, uint32{1} << 31)); in IsNegZero()
126 ConstantR0WithType(&b, U32, uint32{1} << 31)); in IsNegZero()
|
| D | loops.cc | 89 return Lt(values[0], ConstantR0WithType(cond_builder, num_iterations_type, in ForEachIndex()
|
| D | arithmetic.cc | 157 XlaOp shift_amount = ConstantR0WithType(builder, output_type, bits_in_type); in ArgMinMax()
|
| D | pooling.cc | 121 auto divisor = ConstantR0WithType(pooled.builder(), dtype, window_size); in AvgPoolDivideByCount()
|
| D | matrix.cc | 58 auto offset = ConstantR0WithType(builder, S32, k); in GetMatrixDiagonal()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | constants_test.cc | 201 auto scalar1 = ConstantR0WithType(&b, F32, 1); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 332 auto start_index = ConstantR0WithType(builder, S32, j * block_size); in SolveWithInvertedDiagonalBlocks()
|