Home
last modified time | relevance | path

Searched refs:NoFragmentationStatsHeap (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.h47 class NoFragmentationStatsHeap; variable
217 const std::unique_ptr<NoFragmentationStatsHeap<HloValue>>
294 class NoFragmentationStatsHeap : public HeapAlgorithm<BufferType> {
298 NoFragmentationStatsHeap() = default;
299 ~NoFragmentationStatsHeap() override = default;
Dheap_simulator.cc66 HeapSimulator::Run(std::make_unique<NoFragmentationStatsHeap<HloValue>>(), in MinimumMemoryForModule()
80 HeapSimulator::Run(std::make_unique<NoFragmentationStatsHeap<HloValue>>(), in MinimumMemoryForComputation()
93 HeapSimulator::Run(std::make_unique<NoFragmentationStatsHeap<HloValue>>(), in MinimumMemoryForComputation()
344 std::make_unique<NoFragmentationStatsHeap<HloValue>>()), in HeapSimulator()
456 void NoFragmentationStatsHeap<BufferType>::Alloc(const BufferType* buffer, in Alloc()
465 void NoFragmentationStatsHeap<BufferType>::AccountForSubcomputationMemory( in AccountForSubcomputationMemory()
494 void NoFragmentationStatsHeap<BufferType>::Free(const BufferType* buffer, in Free()
501 NoFragmentationStatsHeap<BufferType>::Finish() { in Finish()
Dhlo_memory_scheduler_test.cc54 std::make_unique<NoFragmentationStatsHeap<HloValue>>(), in PeakMemoryUseOfEntryComputation()
Dheap_simulator_test.cc982 NoFragmentationStatsHeap<HloValue> heap; in TEST_F()
987 NoFragmentationStatsHeap<HloValue> heap; in TEST_F()
1000 NoFragmentationStatsHeap<HloValue> heap; in TEST_F()