Home
last modified time | relevance | path

Searched refs:CreateGetTupleElement (Results 1 – 25 of 70) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_simplifier_test.cc78 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F()
97 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_F()
121 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_F()
154 element = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
175 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F()
177 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F()
179 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 2)); in TEST_F()
200 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F()
202 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F()
233 HloInstruction::CreateGetTupleElement(scalar_shape_, p0, 0)); in TEST_F()
[all …]
H A Dwhile_loop_invariant_code_motion_test.cc71 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
73 HloInstruction::CreateGetTupleElement(scalar_s32, param, 1)); in TEST_F()
113 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
115 HloInstruction::CreateGetTupleElement(scalar_s32, param, 1)); in TEST_F()
117 HloInstruction::CreateGetTupleElement(scalar_s32, param, 2)); in TEST_F()
185 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
187 HloInstruction::CreateGetTupleElement(scalar_s32, param, 1)); in TEST_F()
224 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
226 HloInstruction::CreateGetTupleElement(scalar_s32, param, 1)); in TEST_F()
263 HloInstruction::CreateGetTupleElement(scalar_s32, param, 0)); in TEST_F()
[all …]
H A Dcopy_insertion_test.cc316 auto gte = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
350 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildConditionComputation()
372 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyComputation()
380 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation()
413 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyComputation2()
423 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation2()
427 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2)); in BuildDependentBodyComputation2()
448 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyOneReadOnlyComputation()
452 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyOneReadOnlyComputation()
485 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildIndependentBodyComputation()
[all …]
H A Dbfloat16_propagation_test.cc218 HloInstruction::CreateGetTupleElement(shape, all_reduce, 0)); in TEST_F()
220 HloInstruction::CreateGetTupleElement(shape, all_reduce, 1)); in TEST_F()
290 HloInstruction::CreateGetTupleElement(xpose->shape(), tuple1, 1)); in TEST_F()
292 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
294 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
336 HloInstruction::CreateGetTupleElement(add1->shape(), tuple, 1)); in TEST_F()
407 HloInstruction::CreateGetTupleElement(shape, p_f1, 0)); in TEST_F()
409 HloInstruction::CreateGetTupleElement(shape, p_f1, 1)); in TEST_F()
522 HloInstruction::CreateGetTupleElement(shape, fusion, 0)); in TEST_F()
524 HloInstruction::CreateGetTupleElement(shape, fusion, 1)); in TEST_F()
[all …]
H A Dhlo_alias_analysis_test.cc189 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 0)); in TEST_F()
191 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_F()
262 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 0)); in TEST_F()
264 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F()
311 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 0)); in TEST_F()
313 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F()
370 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
372 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
395 HloInstruction::CreateGetTupleElement(scalar_shape_, xla_while, 0)); in TEST_F()
397 HloInstruction::CreateGetTupleElement(scalar_shape_, xla_while, 1)); in TEST_F()
[all …]
H A Dhlo_dataflow_analysis_test.cc164 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 0)); in TEST_P()
166 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_P()
222 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1)); in TEST_P()
224 HloInstruction::CreateGetTupleElement(scalar_shape_, gte_tuple, 0)); in TEST_P()
404 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P()
406 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P()
507 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P()
509 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P()
588 HloInstruction::CreateGetTupleElement(scalar_shape_, level0_param, 0)); in TEST_P()
610 HloInstruction::CreateGetTupleElement(scalar_shape_, level2_while, 0)); in TEST_P()
[all …]
H A Dmemory_space_assignment_test.cc784 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
799 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_P()
801 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_P()
877 HloInstruction::CreateGetTupleElement(shape, p, 0)); in TEST_P()
893 HloInstruction::CreateGetTupleElement(shape, p, 1)); in TEST_P()
897 HloInstruction::CreateGetTupleElement(inner_tuple_shape, p, 2)); in TEST_P()
899 HloInstruction::CreateGetTupleElement(shape, p2, 0)); in TEST_P()
1069 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P()
1071 HloInstruction::CreateGetTupleElement(shape, fusion_param, 1)); in TEST_P()
1220 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P()
[all …]
H A Dtuple_util.cc33 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in ExtractPrefix()
52 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in AppendSuffix()
103 HloInstruction::CreateGetTupleElement(subshape, tuple, i)); in ReplaceTupleWith()
166 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in AddGetTupleElements()
H A Dwhile_loop_simplifier.cc113 user, HloInstruction::CreateGetTupleElement(user->shape(), param, in RemoveDeadTupleIndices()
164 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveDeadTupleIndices()
203 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveDeadTupleIndices()
208 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveDeadTupleIndices()
486 auto new_get = comp->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveRepeatedWhileTupleIndicesHelper()
689 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams()
706 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams()
988 HloInstruction::CreateGetTupleElement(subshape, instr, i)); in GetFlatTupleElems()
1041 gtes.push_back(add_new_instr(HloInstruction::CreateGetTupleElement( in TryFlattenNestedTuples()
1206 return add_new_instr(HloInstruction::CreateGetTupleElement( in TryMergeInductionVariables()
H A Dbuffer_assignment_test.cc218 HloInstruction::CreateGetTupleElement(const4->shape(), param, 0)); in BuildWhileConditionComputation()
246 HloInstruction::CreateGetTupleElement(const1->shape(), param, 0)); in BuildWhileBodyComputation()
250 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation()
805 HloInstruction::CreateGetTupleElement(s32_, cond_param, 0)); in TEST_F()
818 HloInstruction::CreateGetTupleElement(s32_, body_param, 0)); in TEST_F()
820 HloInstruction::CreateGetTupleElement(f32vec10_color1, body_param, 1)); in TEST_F()
849 HloInstruction::CreateGetTupleElement(f32vec10_color1, while_op, 1)); in TEST_F()
1262 HloInstruction::CreateGetTupleElement(f32vec100_, tuple, 0)); in TEST_F()
1537 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
1768 HloInstruction::CreateGetTupleElement(scalar_shape, tuple, 0)); in TEST_F()
[all …]
H A Dbfloat16_conversion_folding_test.cc202 HloInstruction::CreateGetTupleElement(f32_shape, tuple, 0)); in TEST_F()
245 HloInstruction::CreateGetTupleElement(f32_shape, crs, 0)); in TEST_F()
247 HloInstruction::CreateGetTupleElement(f32_shape, crs, 1)); in TEST_F()
H A Dhlo_domain_test.cc484 HloInstruction::CreateGetTupleElement( in TEST_F()
491 HloInstruction::CreateGetTupleElement( in TEST_F()
738 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
741 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
H A Dwhile_util.cc88 wide_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in WidenWhileBody()
142 new_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in MakeInstructionsLiveIn()
279 HloInstruction::CreateGetTupleElement(init_values[i]->shape(), in MakeCountedLoop()
H A Dhlo_live_range_test.cc273 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
286 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_F()
288 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_F()
H A Ddynamic_dimension_inference_test.cc164 HloInstruction::CreateGetTupleElement(input_shape, param, 0)); in TEST_F()
753 HloInstruction::CreateGetTupleElement(input_shape, body_param, 0)); in TEST_F()
755 HloInstruction::CreateGetTupleElement(input_shape, body_param, 1)); in TEST_F()
843 HloInstruction::CreateGetTupleElement(input_shape, true_param, 0)); in TEST_F()
845 HloInstruction::CreateGetTupleElement(input_shape, true_param, 1)); in TEST_F()
864 HloInstruction::CreateGetTupleElement(input_shape, false_param, 1)); in TEST_F()
866 HloInstruction::CreateGetTupleElement(input_shape, false_param, 2)); in TEST_F()
H A Dheap_simulator_test.cc52 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
54 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
534 HloInstruction::CreateGetTupleElement(f32scalar_, fusion, 0)); in TEST_F()
537 HloInstruction::CreateGetTupleElement(f32scalar_, fusion, 1)); in TEST_F()
608 HloInstruction::CreateGetTupleElement(f32scalar_, fusion, 1)); in TEST_F()
847 HloInstruction::CreateGetTupleElement(f32scalar_, tuple, 0)); in TEST_F()
853 HloInstruction::CreateGetTupleElement(f32scalar_, tuple, 1)); in TEST_F()
896 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
898 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
H A Dconditional_simplifier.cc137 HloInstruction::CreateGetTupleElement( in TryRemoveUnusedConditionalOperands()
292 cloned_branch->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveUnusedTupleElements()
543 return computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveConditional()
H A Dhlo_memory_scheduler_test.cc232 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
279 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
H A Dtopk_rewriter.cc236 comp->AddInstruction(HloInstruction::CreateGetTupleElement( in TransformToCustomCall()
239 comp->AddInstruction(HloInstruction::CreateGetTupleElement( in TransformToCustomCall()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc245 HloInstruction::CreateGetTupleElement(hlo->shape(), param, 0)); in GetFinalFftUsingCollectivePermute()
247 HloInstruction::CreateGetTupleElement(hlo->shape(), param, 1)); in GetFinalFftUsingCollectivePermute()
249 body_b.AddInstruction(HloInstruction::CreateGetTupleElement( in GetFinalFftUsingCollectivePermute()
252 body_b.AddInstruction(HloInstruction::CreateGetTupleElement( in GetFinalFftUsingCollectivePermute()
255 HloInstruction::CreateGetTupleElement(iteration->shape(), param, 4)); in GetFinalFftUsingCollectivePermute()
318 HloInstruction::CreateGetTupleElement(iteration->shape(), cond_param, 4)); in GetFinalFftUsingCollectivePermute()
334 HloInstruction::CreateGetTupleElement(hlo->shape(), while_loop, 0)); in GetFinalFftUsingCollectivePermute()
H A Dcustom_call_handler.cc143 b_.AddInstruction(HloInstruction::CreateGetTupleElement( in HandleCustomCallTopK()
156 b_.AddInstruction(HloInstruction::CreateGetTupleElement( in HandleCustomCallTopK()
216 b_.AddInstruction(HloInstruction::CreateGetTupleElement( in HandleCustomCallTopK()
220 b_.AddInstruction(HloInstruction::CreateGetTupleElement( in HandleCustomCallTopK()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbuffer_donation_test.cc186 HloInstruction::CreateGetTupleElement(const4->shape(), param, 0)); in BuildWhileConditionComputation()
214 HloInstruction::CreateGetTupleElement(const1->shape(), param, 0)); in BuildWhileBodyComputation()
218 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation()
239 HloInstruction::CreateGetTupleElement(s32_, while0, 0)); in CreateTestModule()
241 HloInstruction::CreateGetTupleElement(f32v1_, while0, 1)); in CreateTestModule()
H A Dmultioutput_fusion_test.cc98 HloInstruction::CreateGetTupleElement(elem_shape2, tuple, 0)); in RunTest2D()
100 HloInstruction::CreateGetTupleElement(elem_shape2, tuple, 1)); in RunTest2D()
162 HloInstruction::CreateGetTupleElement(elem_shape_U8, tuple, 0)); in RunTest1D()
164 HloInstruction::CreateGetTupleElement(elem_shape_F32, tuple, 1)); in RunTest1D()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dwhile_transformer_test.cc45 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildConditionComputation()
62 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildBodyComputation()
69 auto data = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildBodyComputation()
H A Dcusolver_rewriter.cc101 HloInstruction::CreateGetTupleElement(a_shape, custom_call, 0)); in CreateCholesky()
103 HloInstruction::CreateGetTupleElement(info_shape, custom_call, 2)); in CreateCholesky()

123