Home
last modified time | relevance | path

Searched refs:kAdd (Results 1 – 25 of 153) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc561 enum Reducer { kAdd, kMax }; enumerator
585 (param.reducer == kAdd) ? "_add" : "_max"); in R4ReduceWindowTestDataToString()
586 CHECK(param.reducer == kAdd || param.reducer == kMax); in R4ReduceWindowTestDataToString()
633 CHECK(param.reducer == kAdd || param.reducer == kMax); in DoIt()
635 auto computation = reducer == kAdd in DoIt()
648 CHECK(reducer == kAdd || reducer == kMax); in DoIt()
649 auto reduce_func = reducer == kAdd in DoIt()
680 /*reducer=*/kAdd},
689 /*reducer=*/kAdd},
698 /*reducer=*/kAdd},
[all …]
Dbroadcast_simple_test.cc95 case HloOpcode::kAdd: { in ApplyOpToFloats()
310 {{{1, 1, 1}}, {{2, 1, 0}}, {{1, 1, 1}}, HloOpcode::kAdd},
314 {{{3, 4, 5}}, {{2, 1, 0}}, {{1, 1, 1}}, HloOpcode::kAdd},
315 {{{3, 4, 5}}, {{2, 1, 0}}, {{1, 4, 5}}, HloOpcode::kAdd},
316 {{{3, 4, 5}}, {{2, 1, 0}}, {{3, 4, 1}}, HloOpcode::kAdd},
317 {{{3, 4, 5}}, {{2, 1, 0}}, {{3, 1, 5}}, HloOpcode::kAdd},
319 {{{3, 4, 199}}, {{2, 1, 0}}, {{1, 1, 199}}, HloOpcode::kAdd},
487 {{{2, 3}}, {{1, 0}}, {{2, 1}}, {{2, 1}}, HloOpcode::kAdd, HloOpcode::kAdd},
488 {{{2, 3}}, {{1, 0}}, {{2, 1}}, {{1, 3}}, HloOpcode::kAdd, HloOpcode::kAdd},
493 HloOpcode::kAdd,
[all …]
Dfusion_test.cc154 case HloOpcode::kAdd: in ComputeElementwiseAnswerFloat()
208 ShapeUtil::MakeShape(F32, {3, 1}), HloOpcode::kAdd, const0, const1)); in XLA_TEST_F()
220 ShapeUtil::MakeShape(F32, {2, 3}), HloOpcode::kAdd, concat5, negate7)); in XLA_TEST_F()
259 ShapeUtil::MakeShape(F32, {1, 3}), HloOpcode::kAdd, copy1, const2)); in XLA_TEST_F()
321 HloOpcode::kAdd, broadcast, const_array)); in XLA_TEST_F()
601 ShapeUtil::MakeShape(S32, {}), HloOpcode::kAdd, lhs, rhs)); in MakeReduceTestComputation()
709 ShapeUtil::MakeShape(S32, {1}), HloOpcode::kAdd, const1, const0)); in XLA_TEST_F()
711 ShapeUtil::MakeShape(S32, {1}), HloOpcode::kAdd, const1, add1)); in XLA_TEST_F()
713 ShapeUtil::MakeShape(S32, {1}), HloOpcode::kAdd, const1, add2)); in XLA_TEST_F()
715 ShapeUtil::MakeShape(S32, {1}), HloOpcode::kAdd, const1, add3)); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel_test.cc107 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
108 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
206 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
207 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
239 AddTfOp(testing::kAdd, {1, 2}, {3}); // => 3, 2, 2, 2 in TEST_F()
243 AddTfOp(testing::kAdd, {7, 8}, {9}); // => 10 in TEST_F()
247 AddTfOp(testing::kAdd, {11, 12}, {13}); // => 2, 4 in TEST_F()
249 AddTfOp(testing::kAdd, {14, 15}, {16}); // => 6 in TEST_F()
252 AddTfOp(testing::kAdd, {9, 16}, {17}); // => 16 in TEST_F()
322 AddTfOp(testing::kAdd, {0, kOnes}, {3}); in TEST_F()
[all …]
Ddelegate_test.cc56 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
57 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
79 AddTfOp(testing::kAdd, {0, 1}, {2}); in TEST_F()
98 AddTfOp(testing::kAdd, {0, 1}, {2}); in TEST_F()
119 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
120 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
140 AddTfOp(testing::kAdd, {1, 2}, {3}); in TEST_F()
146 AddTfOp(testing::kAdd, {7, 8}, {9}); in TEST_F()
211 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
212 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc109 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
111 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, add0, b)); in TEST_F()
122 ShapeUtil::MakeShape(F32, {4, 4}), HloOpcode::kAdd, dot, dot)); in TEST_F()
161 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
183 ShapeUtil::MakeShape(F32, {4, 4}), HloOpcode::kAdd, dot, dot)); in TEST_F()
207 shape, HloOpcode::kAdd, in TEST_F()
221 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, dot, dot)); in TEST_F()
273 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
275 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, a)); in TEST_F()
277 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, b, b)); in TEST_F()
[all …]
Ddefuser_test.cc53 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1)); in TEST_F()
69 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1)); in TEST_F()
93 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1)); in TEST_F()
121 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1)); in TEST_F()
133 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, constant, div)); in TEST_F()
160 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1)); in TEST_F()
172 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, constant, div)); in TEST_F()
198 HloInstruction::CreateBinary(shape_, HloOpcode::kAdd, param0, param1)); in TEST_F()
Dhlo_rematerialization_test.cc361 HloInstruction::CreateBinary(vec1024_shape_, HloOpcode::kAdd, rng, tanh)); in TEST_F()
363 vec1024_shape_, HloOpcode::kAdd, rng, in TEST_F()
365 vec1024_shape_, HloOpcode::kAdd, exp, add_0)))); in TEST_F()
367 vec1024_shape_, HloOpcode::kAdd, rng, in TEST_F()
369 vec1024_shape_, HloOpcode::kAdd, tanh, add_1)))); in TEST_F()
445 vec1024_shape_, HloOpcode::kAdd, bcast, bcast)); in TEST_F()
449 vec1024_shape_, HloOpcode::kAdd, bcast, call_1)); in TEST_F()
453 vec1024_shape_, HloOpcode::kAdd, bcast, call_2)); in TEST_F()
457 vec1024_shape_, HloOpcode::kAdd, bcast, call_3)); in TEST_F()
604 vec1024_shape_, HloOpcode::kAdd, bcast, bcast)); in TEST_P()
[all …]
Dhlo_instruction_test.cc154 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
191 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
224 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, foo)); in TEST_F()
255 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param0, c0)); in TEST_F()
257 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, c0, param1)); in TEST_F()
259 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, addleft, addright)); in TEST_F()
302 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param0, neg1)); in TEST_F()
304 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, neg1, param1)); in TEST_F()
306 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, addleft, addright)); in TEST_F()
340 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, value)); in TEST_F()
[all …]
Dhlo_graph_dumper_test.cc53 shape, HloOpcode::kAdd, params[0], params[1]))); in TEST_F()
56 shape, HloOpcode::kAdd, sums[i], params[i + 2]))); in TEST_F()
72 if (instr->opcode() == HloOpcode::kAdd) { in TEST_F()
101 if (instruction->opcode() == HloOpcode::kAdd) { in TEST_F()
Dhlo_cse_test.cc60 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
90 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
122 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
168 shape_r0, HloOpcode::kAdd, constants[0], constants[1])); in TEST_F()
171 shape_r0, HloOpcode::kAdd, root, constants[i])); in TEST_F()
500 HloInstruction::CreateBinary(shape_r0, HloOpcode::kAdd, param0, param1)); in TEST_F()
502 HloInstruction::CreateBinary(shape_r0, HloOpcode::kAdd, param0, param1)); in TEST_F()
545 constant->shape(), HloOpcode::kAdd, negate1, exp1)); in TEST_F()
552 constant->shape(), HloOpcode::kAdd, negate2, exp2)); in TEST_F()
587 constant1->shape(), HloOpcode::kAdd, rng1, rng2)); in TEST_F()
[all …]
Dinstruction_fusion_test.cc169 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param0, param1)); in TEST_F()
216 EXPECT_EQ(Count(*module, HloOpcode::kAdd), 1) << module->ToString(); in TEST_F()
248 EXPECT_EQ(Count(*module, HloOpcode::kAdd), 1) << module->ToString(); in TEST_F()
278 EXPECT_EQ(Count(*module, HloOpcode::kAdd), 1) << module->ToString(); in TEST_F()
309 EXPECT_EQ(Count(*module, HloOpcode::kAdd), 2) << module->ToString(); in TEST_F()
349 EXPECT_EQ(Count(*module, HloOpcode::kAdd), 2) << module->ToString(); in TEST_F()
382 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param0, param1)); in TEST_F()
Dbfloat16_conversion_folding_test.cc37 if (hlo.opcode() == HloOpcode::kAdd || in SupportsBF16Operand()
48 if (hlo.opcode() == HloOpcode::kAdd || in SupportsBF16Output()
59 if (hlo.opcode() == HloOpcode::kAdd || hlo.opcode() == HloOpcode::kTuple || in SupportsMixedPrecisions()
96 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, a, b)); in TEST_F()
103 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, convert1, c)); in TEST_F()
226 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, x, y)); in TEST_F()
Dreshape_mover_test.cc52 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
95 root_shape, HloOpcode::kAdd, reshape0, const1));
121 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
148 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
239 root_shape, HloOpcode::kAdd, reshape0, param1));
329 root_shape, HloOpcode::kAdd, reshape0, const1));
372 root_shape, HloOpcode::kAdd, reshape0, reshape1));
400 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
508 shape2, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
513 builder.AddInstruction(HloInstruction::CreateBinary(shape3, HloOpcode::kAdd, in TEST_F()
Dhlo_dce_test.cc58 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
136 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
145 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
149 constant1->shape(), HloOpcode::kAdd, constant1, constant2)); in TEST_F()
241 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, infeed)); in TEST_F()
337 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, param0, param1)); in TEST_F()
378 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, param0, param1)); in TEST_F()
Dhlo_computation_test.cc111 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, map1, map2)); in TEST_F()
197 r0f32_, HloOpcode::kAdd, constant1, constant2)); in TEST_F()
199 r0f32_, HloOpcode::kAdd, constant2, constant3)); in TEST_F()
201 r0f32_, HloOpcode::kAdd, constant1, constant3)); in TEST_F()
221 builder.AddInstruction(HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, in TEST_F()
223 builder.AddInstruction(HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, in TEST_F()
225 builder.AddInstruction(HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, in TEST_F()
426 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, negate, negate)); in TEST_F()
452 r0f32_, HloOpcode::kAdd, dead_negate, dead_negate)); in TEST_F()
477 r0f32_, HloOpcode::kAdd, constant1, constant2)); in TEST_F()
[all …]
Dhlo_subcomputation_unification_test.cc45 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y)); in CreateR0S32AdditionComputation()
57 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, x, y)); in CreateR1S32AdditionComputation()
83 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y)); in TEST_F()
112 HloInstruction::CreateBinary(r0s32_, HloOpcode::kAdd, x, y)); in TEST_F()
Dbfloat16_normalization_test.cc39 if (hlo.opcode() == HloOpcode::kAdd || in SupportsBF16Operand()
54 if (hlo.opcode() == HloOpcode::kAdd || hlo.opcode() == HloOpcode::kReduce || in SupportsBF16Output()
64 if (hlo.opcode() == HloOpcode::kAdd || hlo.opcode() == HloOpcode::kTuple || in SupportsMixedPrecisions()
105 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kAdd, a, b)); in TEST_F()
108 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, add0, c)); in TEST_F()
192 HloInstruction::CreateBinary(bf16_scalar_shape, HloOpcode::kAdd, in TEST_F()
244 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, x, y)); in TEST_F()
Dbatchnorm_expander.cc90 shape, HloOpcode::kAdd, scalar_lhs, scalar_rhs)); in GetOrCreateScalarAddComputation()
278 add_binary(operand_shape, HloOpcode::kAdd, var_broadcasted, epsilon); in HandleBatchNormTraining()
296 auto shifted_normalized = add_binary(operand_shape, HloOpcode::kAdd, in HandleBatchNormTraining()
386 add_binary(operand_shape, HloOpcode::kAdd, var_broadcasted, epsilon); in HandleBatchNormInference()
405 operand_shape, HloOpcode::kAdd, scaled_normalized, offset_broadcasted); in HandleBatchNormInference()
507 add(Rsqrt(add_binary(activation_shape, HloOpcode::kAdd, in HandleBatchNormGrad()
512 add_binary(feature_shape, HloOpcode::kAdd, variance, epsilon_feature), in HandleBatchNormGrad()
558 add_binary(activation_shape, HloOpcode::kAdd, in HandleBatchNormGrad()
Dbuffer_liveness_test.cc165 HloInstruction::CreateBinary(vec_, HloOpcode::kAdd, negate, exp)); in TEST_F()
248 HloInstruction::CreateBinary(vec_, HloOpcode::kAdd, negate, exp)); in TEST_F()
290 HloInstruction::CreateBinary(vec_, HloOpcode::kAdd, negate, exp)); in TEST_F()
330 HloInstruction::CreateBinary(vec_, HloOpcode::kAdd, param, param)); in TEST_F()
499 tuple_element0_shape, HloOpcode::kAdd, tuple_element0, const0)); in TEST_F()
511 tuple_element1_shape, HloOpcode::kAdd, tuple_element1, const1)); in TEST_F()
563 tuple_element0_shape, HloOpcode::kAdd, tuple_element0, const0)); in TEST_F()
573 tuple_element1_shape, HloOpcode::kAdd, tuple_element0, tuple_element1)); in TEST_F()
637 update_shape, HloOpcode::kAdd, update, slice)); in BuildModule()
793 data_shape, HloOpcode::kAdd, gte0, gte1)); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ldg_test.cc48 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, param)); in TEST_F()
70 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, param)); in TEST_F()
109 HloInstruction::CreateBinary(lhs->shape(), HloOpcode::kAdd, lhs, rhs)); in TEST_F()
121 param_shape, HloOpcode::kAdd, param, param)), in TEST_F()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc33 CHECK(add_or_sub_op->type == OperatorType::kAdd || in FuseAddOrSubParamsIntoPrecedingAffine()
67 const OpType optype = (add_or_sub_op->type == OperatorType::kAdd) in FuseAddOrSubParamsIntoPrecedingAffine()
197 if (binary_op->type != OperatorType::kAdd && in Run()
300 if (binary_op->type == OperatorType::kAdd || in Run()
347 if (binary_op->type == OperatorType::kAdd || in Run()
Dfuse_binary_into_following_affine.cc34 CHECK(add_or_sub_op->type == OperatorType::kAdd || in FuseAddOrSubParamsIntoFollowingAffine()
59 if (add_or_sub_op->type == OperatorType::kAdd) { in FuseAddOrSubParamsIntoFollowingAffine()
159 if (binary_op->type != OperatorType::kAdd && in Run()
263 if (binary_op->type == OperatorType::kAdd || in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_support_checker_test.cc44 scalar_shape, HloOpcode::kAdd, param0, param1)); in TEST_F()
59 sparse_shape, HloOpcode::kAdd, param0, param1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_hlo_support_checker_test.cc44 scalar_shape, HloOpcode::kAdd, param0, param1)); in TEST_F()
59 sparse_shape, HloOpcode::kAdd, param0, param1)); in TEST_F()

1234567