| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | custom_call_test.cc | 91 /*operands=*/{Broadcast(ConstantR0WithType(&b, F32, 42.0), {128})}, in TEST_F()
144 Broadcast(ConstantR0WithType(&b, F32, 1), {128}), in TEST_F()
145 Broadcast(ConstantR0WithType(&b, F32, 2), {256}), in TEST_F()
149 Broadcast(ConstantR0WithType(&b, F32, 3), {1024}), in TEST_F()
150 Broadcast(ConstantR0WithType(&b, F32, 4), {8}), in TEST_F()
|
| /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 | constants.h | 34 XlaOp ConstantR0WithType(XlaBuilder* builder, PrimitiveType type, T value) { in ConstantR0WithType() function
95 return ConstantR0WithType(builder, shape.element_type(), value); in ScalarLike()
|
| D | svd_test.cc | 105 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
107 ConstantR0WithType(builder, F32, size); in GetAverageAbsoluteError()
|
| D | prng.cc | 124 XlaOp const32 = ConstantR0WithType(builder, U64, 32); in Uint64ToUint32s()
135 ConstantR0WithType(builder, U64, 32)); in Uint32sToUint64()
507 ShiftRightLogical(dist, ConstantR0WithType(builder, unsigned_type, 1)); in ConvertRandomBitsToUniformInt()
621 xla::ConstantR0WithType(builder, primitive_type, 0.0), in NormalFloatingPointDistribution()
622 xla::ConstantR0WithType(builder, primitive_type, 1.0), in NormalFloatingPointDistribution()
|
| 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 | loops.cc | 89 return Lt(values[0], ConstantR0WithType(cond_builder, num_iterations_type, in ForEachIndex()
|
| D | pooling.cc | 122 auto divisor = ConstantR0WithType(pooled.builder(), dtype, window_size); in AvgPoolDivideByCount()
|
| D | math.cc | 148 ConstantR0WithType(&b, U64, uint64{1} << 63)); in IsNegZero()
151 ConstantR0WithType(&b, U32, uint32{1} << 31)); in IsNegZero()
158 ConstantR0WithType(&b, U32, uint32{1} << 31)); in IsNegZero()
|
| D | matrix.cc | 68 auto b = Iota(builder, S32, m) + ConstantR0WithType(builder, S32, diagonal); in GetDiagonalMask()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops.cc | 86 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniform()
122 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniformFullInt()
168 xla::ConstantR0WithType(builder, rng_primitive_type, 0.0), in Compile()
169 xla::ConstantR0WithType(builder, rng_primitive_type, 1.0)); in Compile()
305 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in Compile()
|
| D | random_ops_util.cc | 30 (u64_seed1 << ConstantR0WithType(seed0.builder(), xla::U64, 32)); in GetU64FromS32Seeds()
|
| D | roll_op.cc | 73 xla::XlaOp axis_size = xla::ConstantR0WithType( in Compile()
|
| D | stateful_random_ops.cc | 233 xla::ConstantR0WithType(builder, rng_primitive_type, 0.0), in Compile()
234 xla::ConstantR0WithType(builder, rng_primitive_type, 1.0)); in Compile()
414 delta = delta * ConstantR0WithType(delta.builder(), xla::U64, 256); in IncreaseCounter()
|
| D | stateless_random_ops_v2.cc | 231 xla::ConstantR0WithType(builder, rng_primitive_type, 0.0), in Compile()
232 xla::ConstantR0WithType(builder, rng_primitive_type, 1.0)); in Compile()
|
| D | strided_slice_op.cc | 156 xla::ConstantR0WithType(ctx->builder(), ctx->InputXlaType("begin"), in EmitDynamicSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | constants_test.cc | 201 auto scalar1 = ConstantR0WithType(&b, F32, 1); in TEST_F()
|