Home
last modified time | relevance | path

Searched refs:CreateTuple (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_simplifier_test.cc67 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1, param2})); in TEST_F()
97 HloInstruction::CreateTuple({param0, param1, param2})); in TEST_F()
121 HloInstruction::CreateTuple({element, element, element})); in TEST_F()
151 HloInstruction::CreateTuple({nested_tuple, nested_tuple})); in TEST_F()
183 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
208 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
241 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
256 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1, gte2})); in TEST_F()
273 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
279 builder.AddInstruction(HloInstruction::CreateTuple({gte2, gte3})); in TEST_F()
Dcopy_insertion_test.cc93 builder.AddInstruction(HloInstruction::CreateTuple({x})); in TEST_F()
113 builder.AddInstruction(HloInstruction::CreateTuple({constant})); in TEST_F()
182 builder.AddInstruction(HloInstruction::CreateTuple({constant2, x, add})); in TEST_F()
207 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
209 HloInstruction::CreateTuple({constant3, constant2})); in TEST_F()
284 builder.AddInstruction(HloInstruction::CreateTuple({bitcast})); in TEST_F()
379 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
381 HloInstruction::CreateTuple({constant2, constant1})); in TEST_F()
458 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in BuildDependentBodyComputation()
498 builder.AddInstruction(HloInstruction::CreateTuple({add0, data1, data2})); in BuildDependentBodyComputation2()
[all …]
Dhlo_alias_analysis_test.cc154 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_F()
204 HloInstruction::CreateTuple({param0, param1, param0})); in TEST_F()
239 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
285 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
349 HloInstruction::CreateTuple({body_element_0, add})); in TEST_F()
376 builder.AddInstruction(HloInstruction::CreateTuple({negate_1, negate_2})); in TEST_F()
537 HloInstruction::CreateTuple({body_element_0, add})); in TEST_F()
555 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
632 HloInstruction::CreateTuple({body_element_0, add})); in TEST_F()
649 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
[all …]
Dtuple_points_to_analysis_test.cc130 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
183 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
188 HloInstruction::CreateTuple({inner_tuple, constant3})); in TEST_F()
244 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
249 HloInstruction::CreateTuple({inner_tuple, constant3})); in TEST_F()
290 HloInstruction::CreateTuple({constant, constant, constant})); in TEST_F()
313 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
387 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
389 HloInstruction::CreateTuple({constant2, constant2})); in TEST_F()
465 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
[all …]
Dwhile_loop_invariant_code_motion_test.cc78 HloInstruction::CreateTuple({gte_0, gte_1, add_result})); in TEST_F()
137 HloInstruction::CreateTuple({gte_0, gte_1, divide_result})); in TEST_F()
191 builder.AddInstruction(HloInstruction::CreateTuple({gte_0, add_result})); in TEST_F()
231 HloInstruction::CreateTuple({gte_1, gte_0, add_result})); in TEST_F()
271 HloInstruction::CreateTuple({gte_0, gte_1, out_token})); in TEST_F()
281 HloInstruction::CreateTuple({scalar_param, scalar_param, token})); in TEST_F()
324 HloInstruction::CreateTuple({gte_0, gte_1, out_token})); in TEST_F()
334 HloInstruction::CreateTuple({scalar_param, scalar_param, token})); in TEST_F()
376 HloInstruction::CreateTuple({gte_0, gte_1, add_inst})); in TEST_F()
425 HloInstruction::CreateTuple({gte_0, gte_1, add_result})); in TEST_F()
Dtuple_util.cc38 HloInstruction::CreateTuple(tuple_elements)); in ExtractPrefix()
58 HloInstruction::CreateTuple(tuple_elements)); in AppendSuffix()
Dbfloat16_propagation_test.cc283 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1, add2})); in TEST_F()
285 builder.AddInstruction(HloInstruction::CreateTuple({tuple0, xpose})); in TEST_F()
299 builder.AddInstruction(HloInstruction::CreateTuple({dot, add2})); in TEST_F()
332 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in TEST_F()
367 builder.AddInstruction(HloInstruction::CreateTuple({add, dot})); in TEST_F()
395 builder_f0.AddInstruction(HloInstruction::CreateTuple({a_f0, b_f0})); in TEST_F()
484 builder_f.AddInstruction(HloInstruction::CreateTuple({a_f, add_f})); in TEST_F()
531 builder.AddInstruction(HloInstruction::CreateTuple({param, add0})); in TEST_F()
533 builder.AddInstruction(HloInstruction::CreateTuple({param, add1})); in TEST_F()
696 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc152 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_P()
208 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
210 HloInstruction::CreateTuple({tuple, tuple, constant1})); in TEST_P()
496 HloInstruction::CreateTuple({body_element_0, add})); in TEST_P()
514 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
599 HloInstruction::CreateTuple({body_element_0, add})); in TEST_P()
616 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
681 HloInstruction::CreateTuple({inner_element_0, add})); in TEST_P()
696 HloInstruction::CreateTuple({negate, outer_element_1})); in TEST_P()
708 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
[all …]
Dwhile_loop_simplifier.cc244 while_body_root, HloInstruction::CreateTuple(new_while_body_root_elems)); in TryRemoveDeadWhileParams()
260 HloInstruction::CreateTuple(new_while_init_elems)); in TryRemoveDeadWhileParams()
302 computation->AddInstruction(HloInstruction::CreateTuple(new_tuple_elems)); in TryRemoveDeadWhileParams()
374 return HloInstruction::CreateTuple(tuple_elems); in TryRemoveConstantParams()
392 return HloInstruction::CreateTuple(tuple_elems); in TryRemoveConstantParams()
616 return HloInstruction::CreateTuple(elems); in UnflattenTupleInstr()
697 return HloInstruction::CreateTuple(GetFlatTupleElems(instr, &new_instrs)); in TryFlattenNestedTuples()
881 return HloInstruction::CreateTuple(tuple_elems); in TryMergeInductionVariables()
909 return HloInstruction::CreateTuple(tuple_elems); in TryMergeInductionVariables()
925 return add_new_instr(HloInstruction::CreateTuple(tuple_elems)); in TryMergeInductionVariables()
Dbuffer_assignment_test.cc227 builder.AddInstruction(HloInstruction::CreateTuple({addc, addv})); in BuildWhileBodyComputation()
390 HloInstruction::CreateTuple({negate, param0, constant})); in TEST_F()
863 builder.AddInstruction(HloInstruction::CreateTuple({const3, const4})); in TEST_F()
1039 builder.AddInstruction(HloInstruction::CreateTuple({negate, broadcast})); in TEST_F()
1069 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({negate})); in TEST_F()
1076 builder.AddInstruction(HloInstruction::CreateTuple({tuple, broadcast})); in TEST_F()
1181 builder.AddInstruction(HloInstruction::CreateTuple({broadcast})); in TEST_F()
1393 sub_builder.AddInstruction(HloInstruction::CreateTuple({sub_param})); in TEST_F()
1453 a_builder.AddInstruction(HloInstruction::CreateTuple({a_param})); in TEST_F()
1558 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({param})); in TEST_F()
[all …]
Dbuffer_liveness_test.cc370 builder.AddInstruction(HloInstruction::CreateTuple({negate})); in TEST_F()
374 builder.AddInstruction(HloInstruction::CreateTuple({inner_tuple, exp})); in TEST_F()
515 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in TEST_F()
577 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in TEST_F()
647 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in BuildModule()
803 HloInstruction::CreateTuple({gte0, dynamic_update_slice})); in Run()
Dlayout_assignment_test.cc222 HloInstruction::CreateTuple({constant0, constant1})); in TEST_F()
259 HloInstruction::CreateTuple({constant0, constant1})); in TEST_F()
261 HloInstruction::CreateTuple({constant0, constant1})); in TEST_F()
300 builder.AddInstruction(HloInstruction::CreateTuple({constant})); in TEST_F()
302 HloInstruction::CreateTuple({inner_tuple, inner_tuple})); in TEST_F()
488 HloInstruction::CreateTuple({transpose, broadcast2})); in TEST_F()
768 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_F()
780 true_builder.AddInstruction(HloInstruction::CreateTuple({add})); in TEST_F()
796 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
1335 b.AddInstruction(HloInstruction::CreateTuple({add, transpose})); in TEST_F()
Dhlo_cse_test.cc201 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F()
234 builder.AddInstruction(HloInstruction::CreateTuple({exp1, exp2, exp3})); in TEST_F()
440 builder.AddInstruction(HloInstruction::CreateTuple({exp1, exp2})); in TEST_F()
471 builder.AddInstruction(HloInstruction::CreateTuple({exp1, exp2})); in TEST_F()
555 builder.AddInstruction(HloInstruction::CreateTuple({add1, add2})); in TEST_F()
Dbfloat16_conversion_folding_test.cc200 builder.AddInstruction(HloInstruction::CreateTuple({a, convert0})); in TEST_F()
250 HloInstruction::CreateTuple({gte_a, convert_gte_b})); in TEST_F()
Dwhile_util.cc199 body_computation->AddInstruction(HloInstruction::CreateTuple(next_state)); in MakeCountedLoopBodyComputation()
214 HloInstruction::CreateTuple(init_values_with_indvar)); in MakeInitTupleFromInitValues()
Dhlo_memory_scheduler_test.cc164 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F()
211 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({add, mul})); in TEST_F()
Dhlo_instruction_test.cc452 builder.AddInstruction(HloInstruction::CreateTuple({foo, bar, baz, foo})); in TEST_F()
549 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({foo, bar})); in TEST_F()
741 builder.AddInstruction(HloInstruction::CreateTuple({constant, constant})); in TEST_F()
819 builder.AddInstruction(HloInstruction::CreateTuple({sub, sub, mul, c1})); in TEST_F()
900 EXPECT_TRUE(Identical(*HloInstruction::CreateTuple({op1, op2}), in TEST_F()
901 *HloInstruction::CreateTuple({op1, op2}))); in TEST_F()
902 EXPECT_FALSE(Identical(*HloInstruction::CreateTuple({op1, op2}), in TEST_F()
903 *HloInstruction::CreateTuple({op2, op1}))); in TEST_F()
Dhlo_computation_test.cc282 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
329 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
406 builder.AddInstruction(HloInstruction::CreateTuple({token, constant})); in TEST_F()
Dstable_sort_expander.cc123 HloInstruction::CreateTuple(tuple_elements)); in ExpandInstruction()
Dhlo_element_type_converter.cc103 HloInstruction::CreateTuple(tuple_elements)); in ConvertTupleElements()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc80 ? builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})) in BuildBodyComputation()
81 : builder.AddInstruction(HloInstruction::CreateTuple({add1, add0})); in BuildBodyComputation()
99 HloInstruction::CreateTuple({induction_var_init, data_init})) in BuildWhileInstruction()
101 HloInstruction::CreateTuple({data_init, induction_var_init})); in BuildWhileInstruction()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc257 builder.AddInstruction(HloInstruction::CreateTuple({add1, add2})); in TEST_F()
315 HloInstruction::CreateTuple({negate1, negate2, exp2})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ldg_test.cc73 builder.AddInstruction(HloInstruction::CreateTuple({add, square})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc121 HloInstruction::CreateTuple({sub_param, constant, add})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultioutput_fusion_test.cc98 computation->AddInstruction(HloInstruction::CreateTuple({sub, add2})); in RunTest2D()
161 HloInstruction::CreateTuple({sub_U8, add})); in RunTest1D()

12