Searched refs:MemorySpaceAssignment (Results 1 – 3 of 3) sorted by relevance
388 class MemorySpaceAssignment {544 virtual Status Process(MemorySpaceAssignment* memory_space_assignment);602 Status Process(MemorySpaceAssignment* memory_space_assignment) override;791 virtual ~MemorySpaceAssignment() = default;823 MemorySpaceAssignment(HloModule* module, Options options, in MemorySpaceAssignment() function898 MemorySpaceAssignment::MemorySpace destination;944 using MemorySpace = MemorySpaceAssignment::MemorySpace;945 using AllocationValue = MemorySpaceAssignment::AllocationValue;948 MemorySpaceAssignment::AllocationSequence* allocations, in AlternateMemoryBestFitHeap()949 const MemorySpaceAssignment::Options& options, in AlternateMemoryBestFitHeap()[all …]
64 const HloValue& value, const MemorySpaceAssignment::Options& options) { in IsCrossProgramPrefetchCandidate()92 absl::optional<MemorySpaceAssignment::BufferInterval>95 const MemorySpaceAssignment::Options& options) { in FindCrossProgramPrefetchCandidate()96 std::vector<MemorySpaceAssignment::BufferInterval> candidates; in FindCrossProgramPrefetchCandidate()101 MemorySpaceAssignment::BufferInterval interval; in FindCrossProgramPrefetchCandidate()706 bool MemorySpaceAssignment::Allocation::operator==( in operator ==()707 const MemorySpaceAssignment::Allocation& other) const { in operator ==()715 bool MemorySpaceAssignment::CopyAllocation::operator==( in operator ==()716 const MemorySpaceAssignment::CopyAllocation& other) const { in operator ==()724 std::string MemorySpaceAssignment::AllocationValue::ToString() const { in ToString()[all …]
67 MemorySpaceAssignment::GetMemoryBoundednessBufferIntervalCompare( in AssignMemorySpaceUsingCostAnalysis()75 absl::optional<MemorySpaceAssignment::Options> options = absl::nullopt) { in AssignMemorySpace()85 absl::optional<MemorySpaceAssignment::BufferIntervalCompare> in AssignMemorySpace()88 absl::optional<MemorySpaceAssignment::Options> in AssignMemorySpace()122 MemorySpaceAssignment::Options options; in AssignMemorySpace()147 MemorySpaceAssignment::Run(module, *hlo_live_range, *alias_analysis, in AssignMemorySpace()1409 MemorySpaceAssignment::BufferIntervalCompare buffer_interval_compare = in TEST_P()1410 [](const MemorySpaceAssignment::BufferInterval& a, in TEST_P()1411 const MemorySpaceAssignment::BufferInterval& b) { in TEST_P()4011 MemorySpaceAssignment::BufferIntervalCompare buffer_interval_compare = in TEST_P()[all …]