| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | constants_test.cc | 34 ConstantR0WithType(&builder, xla::S32, 4); in XLA_TEST_F()
40 ConstantR0WithType(&builder, xla::S32, 4.5); in XLA_TEST_F()
48 ConstantR0WithType(&builder, xla::F32, -7); in XLA_TEST_F()
50 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 | 122 XlaOp const32 = ConstantR0WithType(builder, U64, 32); in Uint64ToUint32s()
133 ConstantR0WithType(builder, U64, 32)); in Uint32sToUint64()
508 ShiftRightLogical(dist, ConstantR0WithType(builder, unsigned_type, 1)); in ConvertRandomBitsToUniformInt()
622 xla::ConstantR0WithType(builder, primitive_type, 0.0), in NormalFloatingPointDistribution()
623 xla::ConstantR0WithType(builder, primitive_type, 1.0), in NormalFloatingPointDistribution()
|
| D | self_adjoint_eig_test.cc | 114 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
116 ConstantR0WithType(builder, F32, std::max<int64_t>(1, 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_t{1} << 63)); in IsNegZero()
151 ConstantR0WithType(&b, U32, uint32_t{1} << 31)); in IsNegZero()
158 ConstantR0WithType(&b, U32, uint32_t{1} << 31)); in IsNegZero()
|
| D | matrix.cc | 69 auto b = Iota(builder, S32, m) + ConstantR0WithType(builder, S32, diagonal); in GetDiagonalMask()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | custom_call_test.cc | 94 /*operands=*/{Broadcast(ConstantR0WithType(&b, F32, 42.0), {128})}, in TEST_F()
147 Broadcast(ConstantR0WithType(&b, F32, 1), {128}), in TEST_F()
148 Broadcast(ConstantR0WithType(&b, F32, 2), {256}), in TEST_F()
152 Broadcast(ConstantR0WithType(&b, F32, 3), {1024}), in TEST_F()
153 Broadcast(ConstantR0WithType(&b, F32, 4), {8}), in TEST_F()
227 values.push_back(Broadcast(ConstantR0WithType(&b, F32, 1), {128})); in BuildInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops.cc | 87 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniform()
123 xla::XlaOp initial_state = xla::ConstantR0WithType(builder, xla::U64, 0); in StatelessRngUniformFullInt()
169 xla::ConstantR0WithType(builder, rng_primitive_type, 0.0), in Compile()
170 xla::ConstantR0WithType(builder, rng_primitive_type, 1.0)); in Compile()
306 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 | sharding_util_ops.cc | 175 xla::ConstantR0WithType(ctx->builder(), type, /*value=*/0), in CompileInternal()
230 ctx->SetOutput(i, xla::Broadcast(xla::ConstantR0WithType( in CompileInternal()
238 xla::ConstantR0WithType(ctx->builder(), type, /*value=*/0), in CompileInternal()
335 xla::ConstantR0WithType(ctx->builder(), type, /*value=*/0), in CompileInternal()
|
| D | roll_op.cc | 73 xla::XlaOp axis_size = xla::ConstantR0WithType( in Compile()
|
| D | stateful_random_ops.cc | 234 xla::ConstantR0WithType(builder, rng_primitive_type, 0.0), in Compile()
235 xla::ConstantR0WithType(builder, rng_primitive_type, 1.0)); in Compile()
415 delta = delta * ConstantR0WithType(delta.builder(), xla::U64, 256); in IncreaseCounter()
|
| D | stateless_random_ops_v2.cc | 245 xla::ConstantR0WithType(builder, rng_primitive_type, 0.0), in Compile()
246 xla::ConstantR0WithType(builder, rng_primitive_type, 1.0)); in Compile()
|
| D | where_op.cc | 243 XlaOp oob_idx = xla::ConstantR0WithType(b, S32, flattened_size); // ⊥ in CompileWhereWithPrefixSum()
|
| D | strided_slice_op.cc | 163 xla::ConstantR0WithType(ctx->builder(), ctx->InputXlaType("begin"), in EmitDynamicSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bitcast_dtypes_expander.cc | 94 ConstantR0WithType(&b, output_logical_type, input_bit_width), in ExpandInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | constants_test.cc | 201 auto scalar1 = ConstantR0WithType(&b, F32, 1); in TEST_F()
|