Home
last modified time | relevance | path

Searched refs:CreateBinary (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
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()
125 HloInstruction::CreateBinary(shape_, HloOpcode::kSubtract, add, negate)); in TEST_F()
127 HloInstruction::CreateBinary(shape_, HloOpcode::kMultiply, sub, param3)); in TEST_F()
129 HloInstruction::CreateBinary(shape_, HloOpcode::kDivide, mul, param3)); 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()
164 HloInstruction::CreateBinary(shape_, HloOpcode::kSubtract, add, negate)); in TEST_F()
[all …]
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()
121 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
147 sub_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
161 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
182 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
206 rb.AddInstruction(HloInstruction::CreateBinary( 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()
[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_rematerialization_test.cc361 HloInstruction::CreateBinary(vec1024_shape_, HloOpcode::kAdd, rng, tanh)); in TEST_F()
362 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
364 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
366 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
368 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
444 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
448 auto add_2 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
452 auto add_3 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
456 auto add_4 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
603 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
[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()
164 HloInstruction::CreateBinary(r0f32_, HloOpcode::kPower, x, y)); in TEST_F()
Dhlo_cse_test.cc59 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
89 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
121 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
167 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
170 root = builder.AddInstruction(HloInstruction::CreateBinary( 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()
504 HloInstruction::CreateBinary(shape_r0, HloOpcode::kMultiply, add1, add2)); in TEST_F()
544 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
551 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
[all …]
Dbfloat16_conversion_folding_test.cc96 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, a, b)); in TEST_F()
103 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, convert1, c)); in TEST_F()
130 HloInstruction::CreateBinary(f32_shape, HloOpcode::kMultiply, a, b)); in TEST_F()
136 HloInstruction* mul1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
165 HloInstruction::CreateBinary(f32_shape, HloOpcode::kSubtract, a, b)); in TEST_F()
171 HloInstruction* sub1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
225 sum_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
Dhlo_memory_scheduler_test.cc62 HloInstruction::CreateBinary(vec, HloOpcode::kAdd, ab, exp)); in TEST_F()
66 HloInstruction::CreateBinary(vec, HloOpcode::kSubtract, add, negate)); in TEST_F()
172 builder.AddInstruction(HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, in TEST_F()
208 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
210 HloInstruction::CreateBinary(r1f32, HloOpcode::kMultiply, add, c3)); in TEST_F()
220 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, tuple_elm, exp)); in TEST_F()
270 body_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
Dbfloat16_normalization_test.cc105 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kAdd, a, b)); in TEST_F()
108 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, add0, c)); in TEST_F()
133 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, a, b)); in TEST_F()
136 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, mul0, c)); in TEST_F()
163 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, a, b)); in TEST_F()
166 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, sub0, c)); in TEST_F()
192 HloInstruction::CreateBinary(bf16_scalar_shape, HloOpcode::kAdd, in TEST_F()
243 sum_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
Dreshape_mover_test.cc51 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
94 builder.AddInstruction(HloInstruction::CreateBinary(
120 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
147 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
238 builder.AddInstruction(HloInstruction::CreateBinary(
328 builder.AddInstruction(HloInstruction::CreateBinary(
371 builder.AddInstruction(HloInstruction::CreateBinary(
399 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
507 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
513 builder.AddInstruction(HloInstruction::CreateBinary(shape3, HloOpcode::kAdd, in TEST_F()
Dhlo_dce_test.cc57 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
135 auto dead_add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
144 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
148 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
241 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, infeed)); in TEST_F()
336 subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
377 subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
Dwhile_loop_invariant_code_motion_test.cc75 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
120 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
123 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
131 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
134 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
189 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
228 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
373 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
421 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
Dhlo_computation_test.cc111 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, map1, map2)); in TEST_F()
196 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
198 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
200 auto add3 = builder.AddInstruction(HloInstruction::CreateBinary( 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()
451 auto dead_add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
476 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
[all …]
Dbuffer_assignment_test.cc162 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, value)); in BuildMapComputationPlus1()
173 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, param2)); in BuildReduceComputation()
221 auto addc = builder.AddInstruction(HloInstruction::CreateBinary( in BuildWhileBodyComputation()
225 auto addv = builder.AddInstruction(HloInstruction::CreateBinary( in BuildWhileBodyComputation()
361 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, const0, const1)); in TEST_F()
437 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
440 HloInstruction::CreateBinary(f32vec100_, HloOpcode::kAdd, mul, param1)); in TEST_F()
441 auto sub = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
520 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
523 HloInstruction::CreateBinary(f32vec100_, HloOpcode::kAdd, mul, param1)); in TEST_F()
[all …]
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()
498 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
510 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
562 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
572 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
636 update = builder.AddInstruction(HloInstruction::CreateBinary( in BuildModule()
792 gte0 = builder.AddInstruction(HloInstruction::CreateBinary( in Run()
Dalgebraic_simplifier_test.cc65 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, zero)); in TEST_F()
288 HloInstruction::CreateBinary(r0s32, HloOpcode::kMultiply, param0, zero)); in TEST_F()
378 HloInstruction::CreateBinary(scalar_shape, HloOpcode::kAdd, p0, p1)); in TEST_F()
410 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, constant, param0)); in TEST_F()
434 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param0, constant1)); in TEST_F()
436 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, add1, constant2)); in TEST_F()
460 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0)); in TEST_F()
484 HloInstruction::CreateBinary(scalar_shape, HloOpcode::kAdd, p0, p1)); in TEST_F()
543 HloInstruction::CreateBinary(r2f32, HloOpcode::kAdd, bcast, param0)); in TEST_F()
610 HloInstruction::CreateBinary(r0f32, HloOpcode::kSubtract, param0, zero)); in TEST_F()
[all …]
Dbatchnorm_expander.cc89 auto scalar_op = b.AddInstruction(HloInstruction::CreateBinary( in GetOrCreateScalarAddComputation()
108 return HloInstruction::CreateBinary(operand->shape(), HloOpcode::kDivide, in Mean()
126 elements_per_feature_u32 = add_instruction(HloInstruction::CreateBinary( in DynamicElementCountPerFeature()
200 return add(HloInstruction::CreateBinary(shape, opcode, a, b)); in HandleBatchNormTraining()
368 return add(HloInstruction::CreateBinary(shape, opcode, a, b)); in HandleBatchNormInference()
404 auto shifted_normalized = HloInstruction::CreateBinary( in HandleBatchNormInference()
457 return add(HloInstruction::CreateBinary(shape, opcode, a, b)); in HandleBatchNormGrad()
Dcopy_insertion_test.cc145 HloInstruction* add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
179 HloInstruction* add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
447 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildDependentBodyComputation()
455 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildDependentBodyComputation()
487 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildDependentBodyComputation2()
525 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildDependentBodyOneReadOnlyComputation()
555 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildIndependentBodyComputation()
572 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildIndependentBodyComputation()
604 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildNestedBodyComputation()
616 auto add10 = builder.AddInstruction(HloInstruction::CreateBinary( in BuildNestedBodyComputation()
[all …]
/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()
72 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, add, add)); in TEST_F()
109 HloInstruction::CreateBinary(lhs->shape(), HloOpcode::kAdd, lhs, rhs)); in TEST_F()
120 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc63 HloInstruction::CreateBinary(vshape, HloOpcode::kAdd, input1, input2)); in TEST_F()
112 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, two, floor)); in TEST_F()
160 embedded_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
188 HloInstruction::CreateBinary(cshape, HloOpcode::kMultiply, two, floor)); in TEST_F()
251 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, negate, ceil)); in TEST_F()
253 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, ceil, negate)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc113 root_hlo = HloInstruction::CreateBinary(answer_shape, opcode, hlos[1], in TestElementwise2D()
207 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
219 auto add8 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
258 auto add3 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
292 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, x, x)); in XLA_TEST_F()
320 HloInstruction::CreateBinary(ShapeUtil::MakeShape(F32, {2, 3}), in XLA_TEST_F()
600 builder.AddInstruction(HloInstruction::CreateBinary( in MakeReduceTestComputation()
679 nested_builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
708 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
710 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
[all …]
Dmultioutput_fusion_test.cc76 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in RunTest2D()
85 HloInstruction* add2 = builder.AddInstruction(HloInstruction::CreateBinary( in RunTest2D()
87 HloInstruction* sub = builder.AddInstruction(HloInstruction::CreateBinary( in RunTest2D()
140 HloInstruction* add = builder.AddInstruction(HloInstruction::CreateBinary( in RunTest1D()
143 builder.AddInstruction(HloInstruction::CreateBinary( in RunTest1D()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_hlo_support_checker_test.cc43 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
58 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_support_checker_test.cc43 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
58 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
Dcudnn_batchnorm_rewriter.cc130 computation_->AddInstruction(HloInstruction::CreateBinary( in HandleBatchNormTraining()
138 computation_->AddInstruction(HloInstruction::CreateBinary( in HandleBatchNormTraining()
185 computation_->AddInstruction(HloInstruction::CreateBinary( in HandleBatchNormGrad()

123