/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() 481 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyComputation2() 491 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation2() 495 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2)); in BuildDependentBodyComputation2() 516 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in BuildDependentBodyOneReadOnlyComputation() 520 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyOneReadOnlyComputation() [all …]
|
D | bfloat16_propagation_test.cc | 216 HloInstruction::CreateGetTupleElement(shape, all_reduce, 0)); in TEST_F() 218 HloInstruction::CreateGetTupleElement(shape, all_reduce, 1)); in TEST_F() 288 HloInstruction::CreateGetTupleElement(xpose->shape(), tuple1, 1)); in TEST_F() 290 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 292 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 334 HloInstruction::CreateGetTupleElement(add1->shape(), tuple, 1)); in TEST_F() 405 HloInstruction::CreateGetTupleElement(shape, p_f1, 0)); in TEST_F() 407 HloInstruction::CreateGetTupleElement(shape, p_f1, 1)); in TEST_F() 491 HloInstruction::CreateGetTupleElement(shape, fusion, 0)); in TEST_F() 493 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() 281 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 0)); in TEST_F() 283 HloInstruction::CreateGetTupleElement(scalar_shape_, param, 1)); in TEST_F() 343 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F() 345 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F() 368 HloInstruction::CreateGetTupleElement(scalar_shape_, xla_while, 0)); in TEST_F() 370 HloInstruction::CreateGetTupleElement(scalar_shape_, xla_while, 1)); in TEST_F() [all …]
|
D | hlo_dataflow_analysis_test.cc | 154 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 0)); in TEST_P() 156 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1)); in TEST_P() 212 HloInstruction::CreateGetTupleElement(tuple->shape(), nested_tuple, 1)); in TEST_P() 214 HloInstruction::CreateGetTupleElement(scalar_shape_, gte_tuple, 0)); in TEST_P() 490 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P() 492 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P() 593 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P() 595 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P() 675 HloInstruction::CreateGetTupleElement(scalar_shape_, inner_param, 0)); in TEST_P() 677 HloInstruction::CreateGetTupleElement(scalar_shape_, inner_param, 1)); in TEST_P() [all …]
|
D | tuple_util.cc | 33 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in ExtractPrefix() 52 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in AppendSuffix()
|
D | buffer_liveness_test.cc | 450 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 494 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 506 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 558 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 570 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 624 HloInstruction::CreateGetTupleElement(data_shape, tuple_param0, 0)); in BuildModule() 627 HloInstruction::CreateGetTupleElement(data_shape, tuple_param0, 1)); in BuildModule() 782 HloInstruction::CreateGetTupleElement(data_shape, tuple_param0, 0)); in Run() 785 HloInstruction::CreateGetTupleElement(data_shape, tuple_param0, 1)); in Run()
|
D | while_loop_simplifier.cc | 206 user, HloInstruction::CreateGetTupleElement(user->shape(), param, in TryRemoveDeadWhileParams() 256 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveDeadWhileParams() 292 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveDeadWhileParams() 297 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveDeadWhileParams() 370 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams() 387 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams() 633 HloInstruction::CreateGetTupleElement(subshape, instr, i)); in GetFlatTupleElems() 685 gtes.push_back(add_new_instr(HloInstruction::CreateGetTupleElement( in TryFlattenNestedTuples() 847 return add_new_instr(HloInstruction::CreateGetTupleElement( in TryMergeInductionVariables()
|
D | tuple_points_to_analysis_test.cc | 252 HloInstruction::CreateGetTupleElement(inner_tuple->shape(), tuple, 0)); in TEST_F() 637 HloInstruction::CreateGetTupleElement(update_shape, tuple_param0, 1)); in Run() 645 HloInstruction::CreateGetTupleElement(input_shape, tuple_param0, 0)); in Run() 658 HloInstruction::CreateGetTupleElement(starts_shape, tuple_param0, 2)); in Run() 857 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 0)); in TEST_F() 859 HloInstruction::CreateGetTupleElement(elem_shape, tuple, 1)); in TEST_F() 880 HloInstruction::CreateGetTupleElement(data_shape, tuple, 0)); in TEST_F() 882 HloInstruction::CreateGetTupleElement(data_shape, tuple, 1)); in TEST_F() 974 HloInstruction::CreateGetTupleElement(data_shape, tuple, 0)); in TEST_F() 976 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() 244 HloInstruction::CreateGetTupleElement(f32_shape, crs, 0)); in TEST_F() 246 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 | buffer_assignment_test.cc | 192 HloInstruction::CreateGetTupleElement(const4->shape(), param, 0)); in BuildWhileConditionComputation() 220 HloInstruction::CreateGetTupleElement(const1->shape(), param, 0)); in BuildWhileBodyComputation() 224 HloInstruction::CreateGetTupleElement(constv->shape(), param, 1)); in BuildWhileBodyComputation() 1071 HloInstruction::CreateGetTupleElement(f32vec100_, tuple, 0)); in TEST_F() 1309 builder.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F() 1560 HloInstruction::CreateGetTupleElement(scalar_shape, tuple, 0)); in TEST_F() 1878 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 0)); in BuildWhileBodyComputation() 1880 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildWhileBodyComputation() 1948 HloInstruction::CreateGetTupleElement(data_shape_, while0, 2)); in TEST_F() 2165 HloInstruction::CreateGetTupleElement(r0s32, infeed, 0)); in TEST_F() [all …]
|
D | hlo_module_dce.cc | 73 HloInstruction::CreateGetTupleElement( in RunWhileDCE()
|
D | while_util.cc | 84 wide_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in WidenWhileBody() 137 new_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in MakeInstructionsLiveIn()
|
D | hlo_memory_scheduler_test.cc | 167 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F() 214 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
|
D | bfloat16_normalization.cc | 242 HloInstruction::CreateGetTupleElement(*subshape, hlo, i)); in HandleMultipleOutputs() 247 HloInstruction::CreateGetTupleElement(*subshape, hlo, i)); in HandleMultipleOutputs()
|
D | sort_simplifier.cc | 118 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveUnusedOperandFromSort()
|
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() 654 HloInstruction::CreateGetTupleElement(f32scalar_, tuple, 0)); in TEST_F() 660 HloInstruction::CreateGetTupleElement(f32scalar_, tuple, 1)); in TEST_F() 703 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F() 705 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
|
D | hlo_graph_dumper_test.cc | 134 auto gte = b.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
|
D | dynamic_dimension_inference_test.cc | 134 HloInstruction::CreateGetTupleElement(input_shape, param, 0)); in TEST_F() 478 HloInstruction::CreateGetTupleElement(input_shape, body_param, 0)); in TEST_F() 480 HloInstruction::CreateGetTupleElement(input_shape, body_param, 1)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
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 | cudnn_batchnorm_rewriter.cc | 127 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining() 146 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining() 148 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining()
|
D | cudnn_conv_padding_legalization.cc | 337 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in CanonicalizeBackwardInputConvolution() 340 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in CanonicalizeBackwardInputConvolution()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
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()
|