Home
last modified time | relevance | path

Searched refs:GetInPlaceInputOutputPairs (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmulti_output_fusion.cc344 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instr1)) { in LegalToFuseMainConstraints()
348 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instr2)) { in LegalToFuseMainConstraints()
H A Dhlo_alias_analysis.cc173 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instruction)) { in ComputeInPlaceOperationAliasedValues()
187 HloDataflowAnalysis::GetInPlaceInputOutputPairs(use.instruction)) { in ComputeInPlaceOperationAliasedValues()
H A Dhlo_dataflow_analysis.h207 GetInPlaceInputOutputPairs(const HloInstruction* instruction);
H A Dhlo_dataflow_analysis_test.cc3024 auto in_place_pairs = HloDataflowAnalysis::GetInPlaceInputOutputPairs(dus); in TEST_F()
3051 auto in_place_pairs = HloDataflowAnalysis::GetInPlaceInputOutputPairs(fusion); in TEST_F()
3076 auto in_place_pairs = HloDataflowAnalysis::GetInPlaceInputOutputPairs(fusion); in TEST_F()
3108 auto in_place_pairs = HloDataflowAnalysis::GetInPlaceInputOutputPairs(fusion); in TEST_F()
3152 HloDataflowAnalysis::GetInPlaceInputOutputPairs(inner_fusion); in TEST_F()
3157 auto in_place_pairs = HloDataflowAnalysis::GetInPlaceInputOutputPairs(fusion); in TEST_F()
H A Dcopy_insertion.cc867 auto in_place = HloDataflowAnalysis::GetInPlaceInputOutputPairs(instr); in ForceRuntimeOrder()
945 return !HloDataflowAnalysis::GetInPlaceInputOutputPairs(instr).empty(); in InstructionCanIntercept()
1833 HloDataflowAnalysis::GetInPlaceInputOutputPairs(instruction)) { in AddCopiesToResolveInterference()
H A Dhlo_dataflow_analysis.cc1732 auto in_place_pairs = HloDataflowAnalysis::GetInPlaceInputOutputPairs( in GetFusionInstructionInPlaceInputOutputPairs()
1770 HloDataflowAnalysis::GetInPlaceInputOutputPairs( in GetInPlaceInputOutputPairs() function in xla::HloDataflowAnalysis
1875 GetInPlaceInputOutputPairs(user)) { in CanShareOperandBufferWithUser()
H A Dbfloat16_propagation.cc602 HloDataflowAnalysis::GetInPlaceInputOutputPairs(hlo)) { in ResolveInconsistencyOfAliasingBuffersHelper()
H A Dinstruction_fusion.cc906 HloDataflowAnalysis::GetInPlaceInputOutputPairs( in ShouldFuseInPlaceOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_fusible.cc276 if (!HloDataflowAnalysis::GetInPlaceInputOutputPairs(&producer).empty()) { in IsProducerConsumerMultiOutputFusible()