Searched refs:NoFragmentationStatsHeap (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator.h | 46 class NoFragmentationStatsHeap; variable
211 const std::unique_ptr<NoFragmentationStatsHeap<HloValue>>
288 class NoFragmentationStatsHeap : public HeapAlgorithm<BufferType> {
292 NoFragmentationStatsHeap() = default;
293 ~NoFragmentationStatsHeap() override = default;
|
| D | heap_simulator.cc | 61 absl::make_unique<NoFragmentationStatsHeap<HloValue>>(), *module, in MinimumMemoryForModule()
76 absl::make_unique<NoFragmentationStatsHeap<HloValue>>(), computation, in MinimumMemoryForComputation()
90 absl::make_unique<NoFragmentationStatsHeap<HloValue>>(), computation, in MinimumMemoryForComputation()
341 absl::make_unique<NoFragmentationStatsHeap<HloValue>>()), in HeapSimulator()
453 void NoFragmentationStatsHeap<BufferType>::Alloc(const BufferType* buffer, in Alloc()
462 void NoFragmentationStatsHeap<BufferType>::AccountForSubcomputationMemory( in AccountForSubcomputationMemory()
491 void NoFragmentationStatsHeap<BufferType>::Free(const BufferType* buffer, in Free()
498 NoFragmentationStatsHeap<BufferType>::Finish() { in Finish()
|
| D | hlo_memory_scheduler_test.cc | 54 absl::make_unique<NoFragmentationStatsHeap<HloValue>>(), in PeakMemoryUseOfEntryComputation()
|
| D | heap_simulator_test.cc | 983 NoFragmentationStatsHeap<HloValue> heap; in TEST_F()
988 NoFragmentationStatsHeap<HloValue> heap; in TEST_F()
1001 NoFragmentationStatsHeap<HloValue> heap; in TEST_F()
|