Home
last modified time | relevance | path

Searched refs:MemorySpaceAssignmentRepacker (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment_repacking.h25 class MemorySpaceAssignmentRepacker {
27 MemorySpaceAssignmentRepacker(int64_t max_size, int64_t alignment) in MemorySpaceAssignmentRepacker() function
29 virtual ~MemorySpaceAssignmentRepacker() = default;
H A Dmemory_space_assignment_best_fit_repacker.h27 : public MemorySpaceAssignmentRepacker {
34 : MemorySpaceAssignmentRepacker(max_size, alignment), type_(type) {} in MemorySpaceAssignmentRepacker() function
H A Dmemory_space_assignment_best_fit_repacker.cc24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock;
H A Dmemory_space_assignment_best_fit_repacker_test.cc24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock;
H A Dmemory_space_assignment.h1083 MemorySpaceAssignmentRepacker* repacker = nullptr;
1268 : MemorySpaceAssignmentRepacker::AllocationBlock {
1520 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*>&
H A Dmemory_space_assignment.cc1558 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> in Finish()
2255 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> colocations; in AllocateReservedScopedAllocations()
2299 for (MemorySpaceAssignmentRepacker::AllocationBlock* repack_block : in AllocateReservedScopedAllocations()
2541 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*>& allocations) { in ExportAllocationsForRepacking()
2642 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*> colocations; in FinalizeAllocations()
2653 for (MemorySpaceAssignmentRepacker::AllocationBlock* repack_block : in FinalizeAllocations()
H A Dheap_simulator.cc1053 MemorySpaceAssignmentRepacker::AllocationBlock>;
H A Dmemory_space_assignment_test.cc5262 class FakeMemorySpaceAssignmentRepacker : public MemorySpaceAssignmentRepacker {
5268 : MemorySpaceAssignmentRepacker(/*max_size=*/128, /*alignment=*/8), in FakeMemorySpaceAssignmentRepacker()
5529 [](absl::Span<MemorySpaceAssignmentRepacker::AllocationBlock*> in TEST_P()
5587 [&](absl::Span<MemorySpaceAssignmentRepacker::AllocationBlock*> in TEST_P()