Home
last modified time | relevance | path

Searched refs:fusion_param (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc727 auto* fusion_param = GetFusionParameterForOperand(fusion, tuple_param0); in Run() local
729 points_to_analysis_->GetPointsToSet(fusion_param).element({}), in Run()
730 {GetBuffer(fusion_param, {})}); in Run()
732 points_to_analysis_->GetPointsToSet(fusion_param).element({0}), in Run()
733 {GetBuffer(fusion_param, {0})}); in Run()
735 points_to_analysis_->GetPointsToSet(fusion_param).element({1}), in Run()
736 {GetBuffer(fusion_param, {1})}); in Run()
738 points_to_analysis_->GetPointsToSet(fusion_param).element({2}), in Run()
739 {GetBuffer(fusion_param, {2})}); in Run()
742 auto fused_gte0 = GetUniqueFusionParameterUserAt(fusion_param, 0); in Run()
[all …]
Dhlo_dataflow_analysis.cc1203 HloInstruction* fusion_param = in DoesNotUseOperandBuffer() local
1206 GetValueDefinedAt(fusion_param, use.operand_index); in DoesNotUseOperandBuffer()
1301 HloInstruction* fusion_param = in CanShareOperandBufferWithUser() local
1304 GetValueDefinedAt(fusion_param, operand_index); in CanShareOperandBufferWithUser()
1307 return AreTransitiveUsesElementwiseOrTuple(fusion_param); in CanShareOperandBufferWithUser()
Dmemory_space_assignment_test.cc1031 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P() local
1034 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P()
1036 HloInstruction::CreateGetTupleElement(shape, fusion_param, 1)); in TEST_P()
1182 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P() local
1185 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P()
1187 HloInstruction::CreateGetTupleElement(shape, fusion_param, 1)); in TEST_P()
3256 HloInstruction* fusion_param = fusion_builder.AddInstruction( in TEST_P() local
3259 HloInstruction::CreateGetTupleElement(shape, fusion_param, 0)); in TEST_P()
3261 HloInstruction::CreateGetTupleElement(shape, fusion_param, 1)); in TEST_P()