Home
last modified time | relevance | path

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

123

/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()
461 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in BuildDependentBodyComputation()
501 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()
282 builder.AddInstruction(HloInstruction::CreateTuple({gte0, gte1})); in TEST_F()
343 HloInstruction::CreateTuple({body_element_0, add})); in TEST_F()
370 builder.AddInstruction(HloInstruction::CreateTuple({negate_1, negate_2})); in TEST_F()
529 HloInstruction::CreateTuple({body_element_0, add})); in TEST_F()
547 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
624 HloInstruction::CreateTuple({body_element_0, add})); in TEST_F()
641 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
[all …]
Dtuple_points_to_analysis_test.cc132 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
185 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
190 HloInstruction::CreateTuple({inner_tuple, constant3})); in TEST_F()
246 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
251 HloInstruction::CreateTuple({inner_tuple, constant3})); in TEST_F()
292 HloInstruction::CreateTuple({constant, constant, constant})); in TEST_F()
315 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
415 HloInstruction::CreateTuple({constant1, constant2})); in TEST_F()
417 HloInstruction::CreateTuple({constant2, constant2})); in TEST_F()
493 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()
Dhlo_dataflow_analysis_test.cc155 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_P()
211 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
213 HloInstruction::CreateTuple({tuple, tuple, constant1})); in TEST_P()
403 HloInstruction::CreateTuple({body_element_0, add})); in TEST_P()
421 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
506 HloInstruction::CreateTuple({body_element_0, add})); in TEST_P()
523 HloInstruction::CreateTuple({constant1, constant2})); in TEST_P()
583 HloInstruction::CreateTuple({level0_element_0})); in TEST_P()
606 level2_builder.AddInstruction(HloInstruction::CreateTuple({negate})); in TEST_P()
613 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({constant1})); in TEST_P()
[all …]
Dbfloat16_propagation_test.cc284 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1, add2})); in TEST_F()
286 builder.AddInstruction(HloInstruction::CreateTuple({tuple0, xpose})); in TEST_F()
300 builder.AddInstruction(HloInstruction::CreateTuple({dot, add2})); in TEST_F()
333 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in TEST_F()
368 builder.AddInstruction(HloInstruction::CreateTuple({add, dot})); in TEST_F()
396 builder_f0.AddInstruction(HloInstruction::CreateTuple({a_f0, b_f0})); in TEST_F()
514 builder_f.AddInstruction(HloInstruction::CreateTuple({a_f, add_f})); in TEST_F()
561 builder.AddInstruction(HloInstruction::CreateTuple({param, add0})); in TEST_F()
563 builder.AddInstruction(HloInstruction::CreateTuple({param, add1})); in TEST_F()
726 builder.AddInstruction(HloInstruction::CreateTuple({add0, add1})); in TEST_F()
[all …]
Dwhile_loop_simplifier.cc141 while_body_root, HloInstruction::CreateTuple(new_while_body_root_elems)); in RemoveDeadTupleIndices()
157 HloInstruction::CreateTuple(new_while_init_elems)); in RemoveDeadTupleIndices()
200 computation->AddInstruction(HloInstruction::CreateTuple(new_tuple_elems)); in RemoveDeadTupleIndices()
555 return HloInstruction::CreateTuple(tuple_elems); in TryRemoveConstantParams()
573 return HloInstruction::CreateTuple(tuple_elems); in TryRemoveConstantParams()
828 return HloInstruction::CreateTuple(elems); in UnflattenTupleInstr()
909 return HloInstruction::CreateTuple(GetFlatTupleElems(instr, &new_instrs)); in TryFlattenNestedTuples()
1093 return HloInstruction::CreateTuple(tuple_elems); in TryMergeInductionVariables()
1121 return HloInstruction::CreateTuple(tuple_elems); in TryMergeInductionVariables()
1137 return add_new_instr(HloInstruction::CreateTuple(tuple_elems)); in TryMergeInductionVariables()
Dhlo_live_range_test.cc160 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({mul, add})); in TEST_F()
199 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({mul, add})); in TEST_F()
307 HloInstruction::CreateTuple({body_data_next, body_iter_next})); in TEST_F()
317 builder.AddInstruction(HloInstruction::CreateTuple({data, iter})); in TEST_F()
Dbuffer_assignment_test.cc234 builder.AddInstruction(HloInstruction::CreateTuple({addc, addv})); in BuildWhileBodyComputation()
397 HloInstruction::CreateTuple({negate, param0, constant})); in TEST_F()
814 HloInstruction::CreateTuple({body_iter_next, body_data_next})); in TEST_F()
826 builder.AddInstruction(HloInstruction::CreateTuple({iter, negate})); in TEST_F()
1035 builder.AddInstruction(HloInstruction::CreateTuple({const3, const4})); in TEST_F()
1211 builder.AddInstruction(HloInstruction::CreateTuple({negate, broadcast})); in TEST_F()
1241 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({negate})); in TEST_F()
1248 builder.AddInstruction(HloInstruction::CreateTuple({tuple, broadcast})); in TEST_F()
1353 builder.AddInstruction(HloInstruction::CreateTuple({broadcast})); in TEST_F()
1565 sub_builder.AddInstruction(HloInstruction::CreateTuple({sub_param})); in TEST_F()
[all …]
Dconditional_simplifier.cc191 HloInstruction::CreateTuple(selects)); in TryRemoveConditional()
276 HloInstruction::CreateTuple(new_tuple_operands)); in TryRemoveUnusedConditionalOperands()
306 branch_computation->AddInstruction(HloInstruction::CreateTuple({})); in ReplaceRootWithEmptyTupleIfNoUsers()
431 HloInstruction::CreateTuple(new_tuple_root_operands)); in RemoveUnusedTupleElements()
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()
765 builder.AddInstruction(HloInstruction::CreateTuple({param0, param1})); in TEST_F()
777 true_builder.AddInstruction(HloInstruction::CreateTuple({add})); in TEST_F()
793 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed_data})); in TEST_F()
1329 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()
Dconditional_canonicalizer.cc31 branch->AddInstruction(HloInstruction::CreateTuple({root})); in CanonicalizeNonTupleConditional()
Dhlo_dce_test.cc82 builder.AddInstruction(HloInstruction::CreateTuple({})); in TEST_F()
103 builder.AddInstruction(HloInstruction::CreateTuple({})); in TEST_F()
120 builder.AddInstruction(HloInstruction::CreateTuple({})); in TEST_F()
Dhlo_instruction_test.cc454 builder.AddInstruction(HloInstruction::CreateTuple({foo, bar, baz, foo})); in TEST_F()
551 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({foo, bar})); in TEST_F()
743 builder.AddInstruction(HloInstruction::CreateTuple({constant, constant})); in TEST_F()
761 builder.AddInstruction(HloInstruction::CreateTuple({constant, constant})); in TEST_F()
781 builder.AddInstruction(HloInstruction::CreateTuple({constant, constant})); in TEST_F()
801 builder.AddInstruction(HloInstruction::CreateTuple({constant, constant})); in TEST_F()
879 builder.AddInstruction(HloInstruction::CreateTuple({sub, sub, mul, c1})); in TEST_F()
960 EXPECT_TRUE(Identical(*HloInstruction::CreateTuple({op1, op2}), in TEST_F()
961 *HloInstruction::CreateTuple({op1, op2}))); in TEST_F()
962 EXPECT_FALSE(Identical(*HloInstruction::CreateTuple({op1, op2}), in TEST_F()
[all …]
Dhlo_memory_scheduler_test.cc230 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F()
277 auto tuple = builder.AddInstruction(HloInstruction::CreateTuple({add, mul})); in TEST_F()
/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()
Dhorizontal_loop_fusion.cc340 HloInstruction::CreateTuple(std::vector<HloInstruction*>{})); in CreateFusedComputation()
419 auto tuple = comp->AddInstruction(HloInstruction::CreateTuple(output_slices)); in CreateFusedComputation()
460 HloInstruction::CreateTuple(bitcasts)); in Fuse()
Dhorizontal_loop_fusion_test.cc223 builder.AddInstruction(HloInstruction::CreateTuple(var_outs)); in TEST_F()
359 builder.AddInstruction(HloInstruction::CreateTuple(all_outputs)); in TEST_F()
Dcudnn_batchnorm_rewriter.cc198 std::unique_ptr<HloInstruction> replacing_tuple = HloInstruction::CreateTuple( in HandleBatchNormTraining()
287 std::unique_ptr<HloInstruction> replacing_tuple = HloInstruction::CreateTuple( in HandleBatchNormGrad()
Dhorizontal_input_fusion_test.cc130 builder.AddInstruction(HloInstruction::CreateTuple(var_outs)); 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()
/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()

123