Searched refs:AllocationBlock (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_best_fit_repacker.cc | 24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock; typedef
25 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()
|
| D | memory_space_assignment_best_fit_repacker_test.cc | 24 using AllocationBlock = MemorySpaceAssignmentRepacker::AllocationBlock; typedef in xla::MemorySpaceAssignmentBestFitRepackerTest
28 AllocationBlock* MakeAllocationBlock(int64_t start_time, int64_t end_time, in MakeAllocationBlock()
39 AllocationBlock* block = &allocation_blocks_.back(); in MakeAllocationBlock()
44 std::list<AllocationBlock> allocation_blocks_;
49 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()
59 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()
75 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()
92 std::vector<AllocationBlock*> allocation_blocks; in TEST_F()
|
| D | memory_space_assignment_repacking.h | 38 struct AllocationBlock { struct
45 std::vector<AllocationBlock*> colocations; argument
55 bool operator<(const AllocationBlock& other) const { return id < other.id; } argument
61 virtual StatusOr<bool> Repack(absl::Span<AllocationBlock*> allocations) = 0;
|
| D | memory_space_assignment_best_fit_repacker.h | 29 using Type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::Type;
33 Type type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::kTemporal)
36 StatusOr<bool> Repack(absl::Span<AllocationBlock*> allocations) override;
|
| D | memory_space_assignment.h | 1268 : MemorySpaceAssignmentRepacker::AllocationBlock {
1520 std::vector<MemorySpaceAssignmentRepacker::AllocationBlock*>&
|
| D | memory_space_assignment.cc | 1558 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()
|
| D | memory_space_assignment_test.cc | 5266 std::function<void(absl::Span<AllocationBlock*>)> check_fun = nullptr, in FakeMemorySpaceAssignmentRepacker()
5273 StatusOr<bool> Repack(absl::Span<AllocationBlock*> allocations) override { in Repack()
5275 for (AllocationBlock* block : allocations) { in Repack()
5278 for (const AllocationBlock* colocation : block->colocations) { in Repack()
5293 for (AllocationBlock* colocation : block->colocations) { in Repack()
5311 std::function<void(absl::Span<AllocationBlock*>)> check_fun_;
5529 [](absl::Span<MemorySpaceAssignmentRepacker::AllocationBlock*> in TEST_P()
5587 [&](absl::Span<MemorySpaceAssignmentRepacker::AllocationBlock*> in TEST_P()
|
| D | heap_simulator.cc | 1053 MemorySpaceAssignmentRepacker::AllocationBlock>;
|
| /external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/Transforms/Utils/ |
| D | CodeExtractor.h | 98 BasicBlock *AllocationBlock; variable
136 BasicBlock *AllocationBlock = nullptr,
|
| /external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Utils/ |
| D | CodeExtractor.cpp | 248 BasicBlock *AllocationBlock, std::string Suffix) in CodeExtractor() argument
250 BPI(BPI), AC(AC), AllocationBlock(AllocationBlock), in CodeExtractor()
260 BPI(BPI), AC(AC), AllocationBlock(nullptr), AllowVarArgs(false), in CodeExtractor()
1199 AllocationBlock ? &*AllocationBlock->getFirstInsertionPt() in emitCallAndSwitchStatement()
|