/external/tensorflow/tensorflow/compiler/xla/service/ |
D | tuple_simplifier_test.cc | 80 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F() 99 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_F() 123 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_F() 156 element = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 177 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F() 179 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F() 181 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 2)); in TEST_F() 202 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0)); in TEST_F() 204 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1)); in TEST_F() 235 HloInstruction::CreateGetTupleElement(scalar_shape_, p0, 0)); in TEST_F() [all …]
|
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 …]
|
D | copy_insertion_test.cc | 351 auto gte = builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 388 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 421 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildConditionComputation() 443 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyComputation() 451 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation() 484 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyComputation2() 494 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation2() 498 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2)); in BuildDependentBodyComputation2() 519 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyOneReadOnlyComputation() 523 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyOneReadOnlyComputation() [all …]
|
D | bfloat16_propagation_test.cc | 217 HloInstruction::CreateGetTupleElement(shape, all_reduce, 0)); in TEST_F() 219 HloInstruction::CreateGetTupleElement(shape, all_reduce, 1)); in TEST_F() 289 HloInstruction::CreateGetTupleElement(xpose->shape(), tuple1, 1)); in TEST_F() 291 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 293 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 335 HloInstruction::CreateGetTupleElement(add1->shape(), tuple, 1)); in TEST_F() 406 HloInstruction::CreateGetTupleElement(shape, p_f1, 0)); in TEST_F() 408 HloInstruction::CreateGetTupleElement(shape, p_f1, 1)); in TEST_F() 521 HloInstruction::CreateGetTupleElement(shape, fusion, 0)); in TEST_F() 523 HloInstruction::CreateGetTupleElement(shape, fusion, 1)); in TEST_F() [all …]
|
D | hlo_alias_analysis_test.cc | 156 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 0)); in TEST_F() 158 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_F() 229 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 0)); in TEST_F() 231 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F() 278 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 0)); in TEST_F() 280 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F() 337 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F() 339 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F() 362 HloInstruction::CreateGetTupleElement(scalar_shape_, xla_while, 0)); in TEST_F() 364 HloInstruction::CreateGetTupleElement(scalar_shape_, xla_while, 1)); in TEST_F() [all …]
|
D | hlo_dataflow_analysis_test.cc | 157 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 0)); in TEST_P() 159 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_P() 215 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1)); in TEST_P() 217 HloInstruction::CreateGetTupleElement(scalar_shape_, gte_tuple, 0)); in TEST_P() 397 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P() 399 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P() 500 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P() 502 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P() 581 HloInstruction::CreateGetTupleElement(scalar_shape_, level0_param, 0)); in TEST_P() 603 HloInstruction::CreateGetTupleElement(scalar_shape_, level2_while, 0)); in TEST_P() [all …]
|
D | tuple_util.cc | 33 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in ExtractPrefix() 52 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in AppendSuffix()
|
D | memory_space_assignment_test.cc | 749 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P() 764 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_P() 766 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_P() 842 HloInstruction::CreateGetTupleElement(shape, p, 0)); in TEST_P() 858 HloInstruction::CreateGetTupleElement(shape, p, 1)); in TEST_P() 862 HloInstruction::CreateGetTupleElement(inner_tuple_shape, p, 2)); in TEST_P() 864 HloInstruction::CreateGetTupleElement(shape, p2, 0)); in TEST_P() 1034 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P() 1036 HloInstruction::CreateGetTupleElement(shape, fusion_param, 1)); in TEST_P() 1185 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P() [all …]
|
D | while_loop_simplifier.cc | 103 user, HloInstruction::CreateGetTupleElement(user->shape(), param, in RemoveDeadTupleIndices() 153 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveDeadTupleIndices() 190 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveDeadTupleIndices() 195 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveDeadTupleIndices() 349 auto new_get = comp->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveRepeatedWhileTupleIndicesHelper() 551 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams() 568 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams() 845 HloInstruction::CreateGetTupleElement(subshape, instr, i)); in GetFlatTupleElems() 897 gtes.push_back(add_new_instr(HloInstruction::CreateGetTupleElement( in TryFlattenNestedTuples() 1059 return add_new_instr(HloInstruction::CreateGetTupleElement( in TryMergeInductionVariables()
|
D | tuple_points_to_analysis_test.cc | 254 HloInstruction::CreateGetTupleElement(inner_tuple->shape(), tuple, 0)); in TEST_F() 659 HloInstruction::CreateGetTupleElement(data_shape, ccall, 0)); in TEST_F() 661 HloInstruction::CreateGetTupleElement(data_shape, ccall, 1)); in TEST_F() 688 HloInstruction::CreateGetTupleElement(update_shape, tuple_param0, 1)); in Run() 696 HloInstruction::CreateGetTupleElement(input_shape, tuple_param0, 0)); in Run() 709 HloInstruction::CreateGetTupleElement(starts_shape, tuple_param0, 2)); in Run() 908 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F() 910 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F() 931 HloInstruction::CreateGetTupleElement(data_shape, tuple, 0)); in TEST_F() 933 HloInstruction::CreateGetTupleElement(data_shape, tuple, 1)); in TEST_F()
|
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()
|
D | hlo_domain_test.cc | 391 HloInstruction::CreateGetTupleElement( in TEST_F() 398 HloInstruction::CreateGetTupleElement( in TEST_F() 645 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 648 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
|
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()
|
D | buffer_assignment_test.cc | 199 HloInstruction::CreateGetTupleElement(const4->shape(), param, 0)); in BuildWhileConditionComputation() 227 HloInstruction::CreateGetTupleElement(const1->shape(), param, 0)); in BuildWhileBodyComputation() 231 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation() 786 HloInstruction::CreateGetTupleElement(s32_, cond_param, 0)); in TEST_F() 799 HloInstruction::CreateGetTupleElement(s32_, body_param, 0)); in TEST_F() 801 HloInstruction::CreateGetTupleElement(f32vec10_color1, body_param, 1)); in TEST_F() 830 HloInstruction::CreateGetTupleElement(f32vec10_color1, while_op, 1)); in TEST_F() 1243 HloInstruction::CreateGetTupleElement(f32vec100_, tuple, 0)); in TEST_F() 1481 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 1732 HloInstruction::CreateGetTupleElement(scalar_shape, tuple, 0)); in TEST_F() [all …]
|
D | topk_rewriter.cc | 161 comp->AddInstruction(HloInstruction::CreateGetTupleElement( in Run() 164 comp->AddInstruction(HloInstruction::CreateGetTupleElement( in Run()
|
D | conditional_canonicalizer.cc | 40 HloInstruction::CreateGetTupleElement(root_shape, new_conditional, 0)); in CanonicalizeNonTupleConditional()
|
D | conditional_simplifier.cc | 168 return computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveConditional() 271 HloInstruction::CreateGetTupleElement( in TryRemoveUnusedConditionalOperands() 426 cloned_branch->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveUnusedTupleElements()
|
D | hlo_module_dce.cc | 77 HloInstruction::CreateGetTupleElement( in RunWhileDCE()
|
D | hlo_memory_scheduler_test.cc | 233 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F() 280 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_batchnorm_rewriter.cc | 172 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining() 190 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining() 200 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining() 279 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormGrad() 289 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormGrad() 291 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormGrad()
|
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()
|
D | cusolver_rewriter.cc | 97 HloInstruction::CreateGetTupleElement(a_shape, custom_call, 0)); in CreateCholesky() 99 HloInstruction::CreateGetTupleElement(info_shape, custom_call, 2)); in CreateCholesky()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | fft_handler.cc | 247 HloInstruction::CreateGetTupleElement(hlo->shape(), param, 0)); in GetFinalFftUsingCollectivePermute() 249 HloInstruction::CreateGetTupleElement(hlo->shape(), param, 1)); in GetFinalFftUsingCollectivePermute() 251 body_b.AddInstruction(HloInstruction::CreateGetTupleElement( in GetFinalFftUsingCollectivePermute() 254 body_b.AddInstruction(HloInstruction::CreateGetTupleElement( in GetFinalFftUsingCollectivePermute() 257 HloInstruction::CreateGetTupleElement(iteration->shape(), param, 4)); in GetFinalFftUsingCollectivePermute() 320 HloInstruction::CreateGetTupleElement(iteration->shape(), cond_param, 4)); in GetFinalFftUsingCollectivePermute() 336 HloInstruction::CreateGetTupleElement(hlo->shape(), while_loop, 0)); in GetFinalFftUsingCollectivePermute()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | buffer_donation_test.cc | 183 HloInstruction::CreateGetTupleElement(const4->shape(), param, 0)); in BuildWhileConditionComputation() 211 HloInstruction::CreateGetTupleElement(const1->shape(), param, 0)); in BuildWhileBodyComputation() 215 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation() 236 HloInstruction::CreateGetTupleElement(s32_, while0, 0)); in CreateTestModule() 238 HloInstruction::CreateGetTupleElement(f32v1_, while0, 1)); in CreateTestModule()
|
D | multioutput_fusion_test.cc | 100 HloInstruction::CreateGetTupleElement(elem_shape2, tuple, 0)); in RunTest2D() 102 HloInstruction::CreateGetTupleElement(elem_shape2, tuple, 1)); in RunTest2D() 164 HloInstruction::CreateGetTupleElement(elem_shape_U8, tuple, 0)); in RunTest1D() 166 HloInstruction::CreateGetTupleElement(elem_shape_F32, tuple, 1)); in RunTest1D()
|