Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_simplifier_test.cc80 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 …]
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 …]
Dcopy_insertion_test.cc351 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 …]
Dbfloat16_propagation_test.cc216 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 …]
Dhlo_alias_analysis_test.cc156 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 …]
Dhlo_dataflow_analysis_test.cc154 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 …]
Dtuple_util.cc33 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in ExtractPrefix()
52 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in AppendSuffix()
Dbuffer_liveness_test.cc450 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()
Dwhile_loop_simplifier.cc206 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()
Dtuple_points_to_analysis_test.cc252 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()
Dbfloat16_conversion_folding_test.cc202 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()
Dhlo_domain_test.cc391 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()
Dbuffer_assignment_test.cc192 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 …]
Dhlo_module_dce.cc73 HloInstruction::CreateGetTupleElement( in RunWhileDCE()
Dwhile_util.cc84 wide_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in WidenWhileBody()
137 new_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in MakeInstructionsLiveIn()
Dhlo_memory_scheduler_test.cc167 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
214 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
Dbfloat16_normalization.cc242 HloInstruction::CreateGetTupleElement(*subshape, hlo, i)); in HandleMultipleOutputs()
247 HloInstruction::CreateGetTupleElement(*subshape, hlo, i)); in HandleMultipleOutputs()
Dsort_simplifier.cc118 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveUnusedOperandFromSort()
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()
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()
Dhlo_graph_dumper_test.cc134 auto gte = b.AddInstruction(HloInstruction::CreateGetTupleElement( in TEST_F()
Ddynamic_dimension_inference_test.cc134 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/
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()
Dcudnn_batchnorm_rewriter.cc127 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining()
146 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining()
148 computation_->AddInstruction(HloInstruction::CreateGetTupleElement( in HandleBatchNormTraining()
Dcudnn_conv_padding_legalization.cc337 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in CanonicalizeBackwardInputConvolution()
340 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in CanonicalizeBackwardInputConvolution()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultioutput_fusion_test.cc100 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()

12