Home
last modified time | relevance | path

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

123

/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()
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 …]
Dbfloat16_propagation_test.cc217 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 …]
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()
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 …]
Dhlo_dataflow_analysis_test.cc157 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 …]
Dtuple_util.cc33 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in ExtractPrefix()
52 computation->AddInstruction(HloInstruction::CreateGetTupleElement( in AppendSuffix()
Dmemory_space_assignment_test.cc749 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 …]
Dwhile_loop_simplifier.cc103 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()
Dtuple_points_to_analysis_test.cc254 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()
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()
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()
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()
Dbuffer_assignment_test.cc199 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 …]
Dtopk_rewriter.cc161 comp->AddInstruction(HloInstruction::CreateGetTupleElement( in Run()
164 comp->AddInstruction(HloInstruction::CreateGetTupleElement( in Run()
Dconditional_canonicalizer.cc40 HloInstruction::CreateGetTupleElement(root_shape, new_conditional, 0)); in CanonicalizeNonTupleConditional()
Dconditional_simplifier.cc168 return computation->AddInstruction(HloInstruction::CreateGetTupleElement( in TryRemoveConditional()
271 HloInstruction::CreateGetTupleElement( in TryRemoveUnusedConditionalOperands()
426 cloned_branch->AddInstruction(HloInstruction::CreateGetTupleElement( in RemoveUnusedTupleElements()
Dhlo_module_dce.cc77 HloInstruction::CreateGetTupleElement( in RunWhileDCE()
Dhlo_memory_scheduler_test.cc233 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
280 HloInstruction::CreateGetTupleElement(r1f32, tuple, 0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc172 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()
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()
Dcusolver_rewriter.cc97 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/
Dfft_handler.cc247 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/
Dbuffer_donation_test.cc183 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()
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()

123