Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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 …]
H A Dbfloat16_propagation_test.cc110 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
112 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, add0, b)); in TEST_F()
122 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
148 sub_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
162 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
183 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
207 rb.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
223 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, dot, dot)); in TEST_F()
275 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, b)); in TEST_F()
277 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, a, a)); in TEST_F()
[all …]
H A Dhlo_live_range_test.cc93 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
118 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
123 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
154 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
159 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
191 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
196 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
198 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
232 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
237 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
[all …]
H A Dmemory_space_assignment_test.cc300 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, p0, tanh)); in CreateEvictAndPrefetchModule()
302 HloInstruction::CreateBinary(shape, HloOpcode::kSubtract, p0, p1)); in CreateEvictAndPrefetchModule()
304 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, p0, p1)); in CreateEvictAndPrefetchModule()
306 HloInstruction::CreateBinary(shape, HloOpcode::kSubtract, p0, p1)); in CreateEvictAndPrefetchModule()
308 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, a, b)); in CreateEvictAndPrefetchModule()
310 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, a, c)); in CreateEvictAndPrefetchModule()
312 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, a, d)); in CreateEvictAndPrefetchModule()
314 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, b, c)); in CreateEvictAndPrefetchModule()
316 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, b, d)); in CreateEvictAndPrefetchModule()
318 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, c, d)); in CreateEvictAndPrefetchModule()
[all …]
H A Dhlo_instruction_test.cc157 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
194 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, bar)); in TEST_F()
227 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, foo, foo)); in TEST_F()
258 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param0, c0)); in TEST_F()
260 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, c0, param1)); in TEST_F()
262 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, addleft, addright)); in TEST_F()
305 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param0, neg1)); in TEST_F()
307 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, neg1, param1)); in TEST_F()
309 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, addleft, addright)); in TEST_F()
343 HloInstruction::CreateBinary(r0f32, HloOpcode::kAdd, param, value)); in TEST_F()
[all …]
H A Dhlo_cse_test.cc60 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
90 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
136 HloInstruction* root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
139 root = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
478 HloInstruction::CreateBinary(shape_r0, HloOpcode::kAdd, param0, param1)); in TEST_F()
480 HloInstruction::CreateBinary(shape_r0, HloOpcode::kAdd, param0, param1)); in TEST_F()
482 HloInstruction::CreateBinary(shape_r0, HloOpcode::kMultiply, add1, add2)); in TEST_F()
522 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
529 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
564 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
[all …]
H A Dbuffer_assignment_test.cc187 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, value)); in BuildMapComputationPlus1()
199 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, param, param2)); in BuildReduceComputation()
247 auto addc = builder.AddInstruction(HloInstruction::CreateBinary( in BuildWhileBodyComputation()
251 auto addv = builder.AddInstruction(HloInstruction::CreateBinary( in BuildWhileBodyComputation()
385 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, const0, const1)); in TEST_F()
461 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
464 HloInstruction::CreateBinary(f32vec100_, HloOpcode::kAdd, mul, param1)); in TEST_F()
465 auto sub = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
543 auto mul = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
546 HloInstruction::CreateBinary(f32vec100_, HloOpcode::kAdd, mul, param1)); in TEST_F()
[all …]
H A 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()
H A Dreshape_mover_test.cc52 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
95 builder.AddInstruction(HloInstruction::CreateBinary(
121 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
148 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
239 builder.AddInstruction(HloInstruction::CreateBinary(
329 builder.AddInstruction(HloInstruction::CreateBinary(
372 builder.AddInstruction(HloInstruction::CreateBinary(
400 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
508 auto add0 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
514 builder.AddInstruction(HloInstruction::CreateBinary(shape3, HloOpcode::kAdd, in TEST_F()
H A Dhlo_computation_test.cc110 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, map1, map2)); in TEST_F()
177 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
179 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
181 auto add3 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
202 builder.AddInstruction(HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, in TEST_F()
204 builder.AddInstruction(HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, in TEST_F()
206 builder.AddInstruction(HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, in TEST_F()
407 HloInstruction::CreateBinary(r0f32_, HloOpcode::kAdd, negate, negate)); in TEST_F()
433 auto dead_add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
458 auto add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
[all …]
H A 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()
H A Dhlo_memory_scheduler_test.cc81 HloInstruction::CreateBinary(vec, HloOpcode::kAdd, ab, exp)); in TEST_F()
85 HloInstruction::CreateBinary(vec, HloOpcode::kSubtract, add, negate)); in TEST_F()
237 builder.AddInstruction(HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, in TEST_F()
273 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, c1, c2)); in TEST_F()
275 HloInstruction::CreateBinary(r1f32, HloOpcode::kMultiply, add, c3)); in TEST_F()
285 HloInstruction::CreateBinary(r1f32, HloOpcode::kAdd, tuple_elm, exp)); in TEST_F()
H A Dhlo_rematerialization_test.cc283 HloInstruction::CreateBinary(vec1024_shape_, HloOpcode::kAdd, rng, tanh)); in TEST_F()
284 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
286 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
288 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
290 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
366 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
370 auto add_2 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
374 auto add_3 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
378 auto add_4 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
523 auto add_1 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_P()
[all …]
H A Dhlo_dce_test.cc57 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
171 auto dead_add = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
180 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
184 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
278 body_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
372 subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
413 subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
460 called_subcomp_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
H A 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()
H A Dbfloat16_normalization_test.cc107 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kAdd, a, b)); in TEST_F()
110 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, add0, c)); in TEST_F()
135 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, a, b)); in TEST_F()
138 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, mul0, c)); in TEST_F()
165 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, a, b)); in TEST_F()
168 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, sub0, c)); in TEST_F()
194 HloInstruction::CreateBinary(bf16_scalar_shape, HloOpcode::kAdd, in TEST_F()
245 sum_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_hlo_schedule_test.cc91 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, x, y)); in TEST_F()
93 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, y, z)); in TEST_F()
95 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, add1, add2)); in TEST_F()
118 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, x, y)); in TEST_F()
120 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, add0, y)); in TEST_F()
122 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, add1, z)); in TEST_F()
142 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, add1, add2)); in TEST_F()
143 HloInstruction* add4 = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
180 reduction_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
195 HloInstruction::CreateBinary(f32_2x2_, HloOpcode::kAdd, x, y)); in TEST_F()
[all …]
H A Dhorizontal_loop_fusion_test.cc218 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
221 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
316 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, grad, grad)); in TEST_F()
318 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, grad_squared, in TEST_F()
321 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
324 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
342 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
345 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
350 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
353 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
[all …]
/aosp_15_r20/external/angle/third_party/spirv-tools/src/test/link/
H A Dbinary_version_test.cpp26 spvtest::Binary CreateBinary(uint32_t version) { in CreateBinary() function
51 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
52 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
64 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
65 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 5)), in TEST_F()
79 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
80 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 5)), in TEST_F()
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/test/link/
Dbinary_version_test.cpp26 spvtest::Binary CreateBinary(uint32_t version) { in CreateBinary() function
51 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
52 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
64 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
65 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 5)), in TEST_F()
79 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
80 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 5)), in TEST_F()
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/test/link/
H A Dbinary_version_test.cpp26 spvtest::Binary CreateBinary(uint32_t version) { in CreateBinary() function
51 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
52 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
64 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
65 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 5)), in TEST_F()
79 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 3)), in TEST_F()
80 CreateBinary(SPV_SPIRV_VERSION_WORD(1, 5)), in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_ldg_test.cc52 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, param)); in TEST_F()
79 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, param)); in TEST_F()
81 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, add, add)); in TEST_F()
123 HloInstruction::CreateBinary(lhs->shape(), HloOpcode::kAdd, lhs, rhs)); in TEST_F()
134 builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc197 HloInstruction::CreateBinary(hlo->shape(), HloOpcode::kMultiply, in GetCorrectionFactor()
201 exp_operand = add_hlo(HloInstruction::CreateBinary( in GetCorrectionFactor()
265 constant_factor = body_b.AddInstruction(HloInstruction::CreateBinary( in GetFinalFftUsingCollectivePermute()
268 constant_factor = body_b.AddInstruction(HloInstruction::CreateBinary( in GetFinalFftUsingCollectivePermute()
276 body_b.AddInstruction(HloInstruction::CreateBinary( in GetFinalFftUsingCollectivePermute()
278 dest_transform = body_b.AddInstruction(HloInstruction::CreateBinary( in GetFinalFftUsingCollectivePermute()
300 i = body_b.AddInstruction(HloInstruction::CreateBinary( in GetFinalFftUsingCollectivePermute()
416 result = b_.AddInstruction(HloInstruction::CreateBinary( in HandleFft()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_fusion_test.cc62 HloInstruction::CreateBinary(vshape, HloOpcode::kAdd, input1, input2)); in TEST_F()
111 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, two, floor)); in TEST_F()
159 embedded_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
187 HloInstruction::CreateBinary(cshape, HloOpcode::kMultiply, two, floor)); in TEST_F()
250 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, negate, ceil)); in TEST_F()
252 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, ceil, negate)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcpu_gpu_fusion_test.cc110 root_hlo = HloInstruction::CreateBinary(answer_shape, opcode, hlos[1], in TestElementwise2D()
204 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
216 auto add8 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
255 auto add3 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
289 HloInstruction::CreateBinary(shape, HloOpcode::kMultiply, x, x)); in XLA_TEST_F()
317 HloInstruction::CreateBinary(ShapeUtil::MakeShape(F32, {2, 3}), in XLA_TEST_F()
597 builder.AddInstruction(HloInstruction::CreateBinary( in MakeReduceTestComputation()
675 nested_builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
704 auto add1 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
706 auto add2 = builder.AddInstruction(HloInstruction::CreateBinary( in XLA_TEST_F()
[all …]

1234