/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | tuple_simplifier_test.cc | 78 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 D | while_loop_invariant_code_motion_test.cc | 71 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 D | copy_insertion_test.cc | 316 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 D | bfloat16_propagation_test.cc | 218 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 D | hlo_alias_analysis_test.cc | 189 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 D | hlo_dataflow_analysis_test.cc | 164 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 D | memory_space_assignment_test.cc | 784 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 D | tuple_util.cc | 33 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 D | while_loop_simplifier.cc | 113 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 D | buffer_assignment_test.cc | 218 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 D | bfloat16_conversion_folding_test.cc | 202 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 D | hlo_domain_test.cc | 484 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 D | while_util.cc | 88 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 D | hlo_live_range_test.cc | 273 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 D | dynamic_dimension_inference_test.cc | 164 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 D | heap_simulator_test.cc | 52 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 D | conditional_simplifier.cc | 137 HloInstruction::CreateGetTupleElement( in TryRemoveUnusedConditionalOperands() 292 cloned_branch->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveUnusedTupleElements() 543 return computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveConditional()
|
H A D | hlo_memory_scheduler_test.cc | 232 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F() 279 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
|
H A D | topk_rewriter.cc | 236 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 D | fft_handler.cc | 245 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 D | custom_call_handler.cc | 143 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 D | buffer_donation_test.cc | 186 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 D | multioutput_fusion_test.cc | 98 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 D | while_transformer_test.cc | 45 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildConditionComputation() 62 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildBodyComputation() 69 auto data = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildBodyComputation()
|
H A D | cusolver_rewriter.cc | 101 HloInstruction::CreateGetTupleElement(a_shape, custom_call, 0)); in CreateCholesky() 103 HloInstruction::CreateGetTupleElement(info_shape, custom_call, 2)); in CreateCholesky()
|