Searched refs:HloAliasAnalysis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis.cc | 344 HloAliasAnalysis::HloAliasAnalysis(HloModule* module) : module_(module) {} in HloAliasAnalysis() function in xla::HloAliasAnalysis
346 const HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
353 HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
355 return GetBuffer(static_cast<const HloAliasAnalysis*>(this) in GetUniqueBufferAt()
360 std::vector<const HloBuffer*> HloAliasAnalysis::ComputeBuffersAt( in ComputeBuffersAt()
375 bool HloAliasAnalysis::InstructionBuffersAreAmbiguous( in InstructionBuffersAreAmbiguous()
392 bool HloAliasAnalysis::InstructionBuffersAreDistinct( in InstructionBuffersAreDistinct()
426 Status HloAliasAnalysis::Verify() const { in Verify()
452 string HloAliasAnalysis::ToString() const { in ToString()
490 StatusOr<std::unique_ptr<HloAliasAnalysis>> HloAliasAnalysis::Run( in Run()
[all …]
|
| D | hlo_alias_analysis.h | 39 class HloAliasAnalysis {
43 static StatusOr<std::unique_ptr<HloAliasAnalysis>> Run(
103 explicit HloAliasAnalysis(HloModule* module);
|
| D | hlo_alias_analysis_test.cc | 49 HloAliasAnalysis& RunAnalysis() { in RunAnalysis()
50 analysis_ = HloAliasAnalysis::Run(module_.get(), in RunAnalysis()
112 std::unique_ptr<HloAliasAnalysis> analysis_;
129 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
164 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
208 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
255 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
301 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
387 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
433 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
[all …]
|
| D | copy_insertion.cc | 256 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
317 Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis, in AddCopiesForConditional()
475 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
966 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddCopiesToResolveInterference()
967 HloAliasAnalysis::Run(module, fusion_can_share_buffer_)); in AddCopiesToResolveInterference()
991 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddSpecialCaseCopies()
992 HloAliasAnalysis::Run(module, fusion_can_share_buffer_)); in AddSpecialCaseCopies()
1093 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in VerifyNoLiveRangeInterference()
1094 HloAliasAnalysis::Run(module, fusion_can_share_buffer_)); in VerifyNoLiveRangeInterference()
1101 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in RemoveUnnecessaryCopies()
[all …]
|