Searched refs:AllocationBlock (Results 1 – 8 of 8) sorted by relevance
24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock; typedef25 using Type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::Type;31 : public GlobalDecreasingSizeBestFitHeap<AllocationBlock> {34 : GlobalDecreasingSizeBestFitHeap<AllocationBlock>(alignment, type), in BestFitRepacker()37 void ImportAllocationBlocks(absl::Span<AllocationBlock*> allocations) { in ImportAllocationBlocks()39 for (AllocationBlock* allocation_block : allocations) { in ImportAllocationBlocks()43 allocation_block->colocations, [&](AllocationBlock* search) { in ImportAllocationBlocks()63 for (AllocationBlock* block : allocation_blocks_) { in Repack()75 absl::Span<AllocationBlock*> allocation_blocks_;81 absl::Span<AllocationBlock*> allocations) { in Repack()
24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock; typedef in xla::MemorySpaceAssignmentBestFitRepackerTest28 AllocationBlock* MakeAllocationBlock(int64 start_time, int64 end_time, in MakeAllocationBlock()37 AllocationBlock* block = &allocation_blocks_.back(); in MakeAllocationBlock()42 std::list<AllocationBlock> allocation_blocks_;47 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()57 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()73 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()
38 struct AllocationBlock { struct45 std::vector<AllocationBlock*> colocations; argument55 bool operator<(const AllocationBlock& other) const { return id < other.id; } argument61 virtual StatusOr<bool> Repack(absl::Span<AllocationBlock*> allocations) = 0;
29 using Type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::Type;33 Type type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::kTemporal)36 StatusOr<bool> Repack(absl::Span<AllocationBlock*> allocations) override;
1007 : MemorySpaceAssignmentRepacker::AllocationBlock {1258 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*>&
1221 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> in Finish()1808 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*>& allocations) { in ExportAllocationsForRepacking()1902 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> colocations; in FinalizeAllocations()1913 for (MemorySpaceAssignmentRepacker::AllocationBlock* repack_block : in FinalizeAllocations()
4381 std::function<void(absl::Span<AllocationBlock*>)> check_fun = nullptr, in FakeMemorySpaceAssignmentRepacker()4388 StatusOr<bool> Repack(absl::Span<AllocationBlock*> allocations) override { in Repack()4390 for (AllocationBlock* block : allocations) { in Repack()4393 for (const AllocationBlock* colocation : block->colocations) { in Repack()4408 for (AllocationBlock* colocation : block->colocations) { in Repack()4426 std::function<void(absl::Span<AllocationBlock*>)> check_fun_;4644 [](absl::Span<MemorySpaceAssignmentRepacker::AllocationBlock*> in TEST_P()
1052 MemorySpaceAssignmentRepacker::AllocationBlock>;