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.cc676 auto* fusion_param = GetFusionParameterForOperand(fusion, tuple_param0); in Run() local
678 points_to_analysis_->GetPointsToSet(fusion_param).element({}), in Run()
679 {GetBuffer(fusion_param, {})}); in Run()
681 points_to_analysis_->GetPointsToSet(fusion_param).element({0}), in Run()
682 {GetBuffer(fusion_param, {0})}); in Run()
684 points_to_analysis_->GetPointsToSet(fusion_param).element({1}), in Run()
685 {GetBuffer(fusion_param, {1})}); in Run()
687 points_to_analysis_->GetPointsToSet(fusion_param).element({2}), in Run()
688 {GetBuffer(fusion_param, {2})}); in Run()
691 auto fused_gte0 = GetUniqueFusionParameterUserAt(fusion_param, 0); in Run()
[all …]
Dhlo_dataflow_analysis.cc934 HloInstruction* fusion_param = in DoesNotUseOperandBuffer() local
937 GetValueDefinedAt(fusion_param, use.operand_index); in DoesNotUseOperandBuffer()
977 auto* fusion_param = fusion_param_value.instruction(); in CanDoInPlaceDynamicUpdateSlice() local
978 CHECK_EQ(fusion_param->opcode(), HloOpcode::kParameter); in CanDoInPlaceDynamicUpdateSlice()
979 CHECK_EQ(fusion_param->parent(), fusion->fused_instructions_computation()); in CanDoInPlaceDynamicUpdateSlice()
1037 HloInstruction* fusion_param = in CanShareOperandBufferWithUser() local
1041 GetValueDefinedAt(fusion_param, operand_index); in CanShareOperandBufferWithUser()
1061 return AreTransitiveUsesElementwiseOrTuple(fusion_param); in CanShareOperandBufferWithUser()
Dtuple_points_to_analysis.cc731 HloInstruction* fusion_param = in CanShareOperandBufferWithUser() local
734 fusion_param); in CanShareOperandBufferWithUser()