Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_repacking.h25 class MemorySpaceAssignmentRepacker {
27 MemorySpaceAssignmentRepacker(int64 max_size, int64 alignment) in MemorySpaceAssignmentRepacker() function
29 virtual ~MemorySpaceAssignmentRepacker() = default;
Dmemory_space_assignment_best_fit_repacker.h27 : public MemorySpaceAssignmentRepacker {
34 : MemorySpaceAssignmentRepacker(max_size, alignment), type_(type) {} in MemorySpaceAssignmentRepacker() function
Dmemory_space_assignment_best_fit_repacker.cc24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock;
Dmemory_space_assignment_best_fit_repacker_test.cc24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock;
Dmemory_space_assignment.h464 MemorySpaceAssignmentRepacker* repacker = nullptr;
1007 : MemorySpaceAssignmentRepacker::AllocationBlock {
1258 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*>&
Dmemory_space_assignment.cc1221 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()
Dheap_simulator.cc1052 MemorySpaceAssignmentRepacker::AllocationBlock>;
Dmemory_space_assignment_test.cc4377 class FakeMemorySpaceAssignmentRepacker : public MemorySpaceAssignmentRepacker {
4383 : MemorySpaceAssignmentRepacker(/*max_size=*/128, /*alignment=*/8), in FakeMemorySpaceAssignmentRepacker()
4644 [](absl::Span<MemorySpaceAssignmentRepacker::AllocationBlock*> in TEST_P()