Searched refs:LogicalBufferAnalysis (Results 1 – 4 of 4) sorted by relevance
44 /* static */ StatusOr<std::unique_ptr<LogicalBufferAnalysis>>45 LogicalBufferAnalysis::Run(const HloModule* module) { in Run()46 std::unique_ptr<LogicalBufferAnalysis> analysis( in Run()47 new LogicalBufferAnalysis(module)); in Run()52 Status LogicalBufferAnalysis::Analyze() { in Analyze()78 LogicalBuffer& LogicalBufferAnalysis::GetBuffer(LogicalBuffer::Id id) const { in GetBuffer()84 LogicalBuffer& LogicalBufferAnalysis::GetBuffer(HloInstruction* instruction, in GetBuffer()89 void LogicalBufferAnalysis::NewLogicalBuffer(HloInstruction* instruction, in NewLogicalBuffer()100 Status LogicalBufferAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()111 Status LogicalBufferAnalysis::HandleGetTupleElement(HloInstruction*) { in HandleGetTupleElement()[all …]
27 class LogicalBufferAnalysis : public DfsHloVisitorWithDefault {30 static StatusOr<std::unique_ptr<LogicalBufferAnalysis>> Run(47 explicit LogicalBufferAnalysis(const HloModule* module) : module_(module) {} in LogicalBufferAnalysis() function
279 std::unique_ptr<LogicalBufferAnalysis> logical_buffer_analysis) in TuplePointsToAnalysis()350 const std::unique_ptr<LogicalBufferAnalysis> logical_buffer_analysis_;
139 auto logical_buffer_analysis = LogicalBufferAnalysis::Run(module); in Run()