Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc33 StatusOr<int64> HeapSimulator::MinimumMemoryForModule( in MinimumMemoryForModule()
50 HeapSimulator::Result result, in MinimumMemoryForModule()
51 HeapSimulator::Run(absl::make_unique<NoFragmentationStatsHeap>(), *module, in MinimumMemoryForModule()
57 StatusOr<int64> HeapSimulator::MinimumMemoryForComputation( in MinimumMemoryForComputation()
64 HeapSimulator::Result result, in MinimumMemoryForComputation()
65 HeapSimulator::Run(absl::make_unique<NoFragmentationStatsHeap>(), in MinimumMemoryForComputation()
67 size_function, HeapSimulator::Options(), in MinimumMemoryForComputation()
73 StatusOr<HeapSimulator::Result> HeapSimulator::Run( in Run()
78 HeapSimulator heap(std::move(algorithm), size_fn, options, &schedule); in Run()
88 StatusOr<HeapSimulator::Result> HeapSimulator::Run( in Run()
[all …]
Dheap_simulator.h47 class HeapSimulator {
140 HeapSimulator(std::unique_ptr<HeapAlgorithm> algorithm,
145 ~HeapSimulator();
213 using Chunk = HeapSimulator::Chunk;
214 using Result = HeapSimulator::Result;
Dheap_simulator_test.cc98 HeapSimulator::MinimumMemoryForModule(schedule, size_fn).ValueOrDie()); in TEST_F()
214 EXPECT_EQ(64, HeapSimulator::MinimumMemoryForComputation( in TEST_F()
276 HeapSimulator::Run(std::move(algorithm), *module_->entry_computation(), in HeapSimulatorTracker()
313 result_ = HeapSimulator::Run(std::move(algorithm), *module_, schedule, in RunWholeModule()
352 HeapSimulator::Result result_;
925 const HeapSimulator::Result result = heap.Finish(); in TEST_F()
941 const HeapSimulator::Result result = heap.Finish(); in TEST_F()
969 const HeapSimulator::Result result = heap.Finish(); in TEST_F()
1013 const HeapSimulator::Result result = heap.Finish(); in TEST_F()
1055 const HeapSimulator::Result result = heap.Finish(); in TEST_F()
[all …]
Dhlo_memory_scheduler.cc528 HeapSimulator::MinimumMemoryForComputation( in DefaultMemoryScheduler()
537 HeapSimulator::MinimumMemoryForComputation( in DefaultMemoryScheduler()
547 HeapSimulator::MinimumMemoryForComputation( in DefaultMemoryScheduler()
584 HeapSimulator::MinimumMemoryForComputation( in ScheduleModule()
Dbuffer_assignment.cc649 HeapSimulator::MinimumMemoryForModule(schedule, buffer_size_)); in ComputeSummaryStats()
1191 HeapSimulator::Options options; in AssignBuffersWithSequentialOrdering()
1197 const HeapSimulator::Result result, in AssignBuffersWithSequentialOrdering()
1198 HeapSimulator::Run(get_heap_algorithm(alignment), in AssignBuffersWithSequentialOrdering()
1222 HeapSimulator::Options options; in AssignBuffersWithSequentialOrdering()
1227 const HeapSimulator::Result result, in AssignBuffersWithSequentialOrdering()
1228 HeapSimulator::Run(get_heap_algorithm(alignment), *computation, in AssignBuffersWithSequentialOrdering()
1323 const HeapSimulator::Result& result, BufferAssignment* assignment, in AssignBuffersFromHeapSimulator()
1340 const HeapSimulator::Chunk& chunk = buffer_chunk.second; in AssignBuffersFromHeapSimulator()
Dhlo_memory_scheduler_test.cc302 EXPECT_EQ(16, HeapSimulator::MinimumMemoryForComputation( in TEST_F()
308 EXPECT_EQ(17, HeapSimulator::MinimumMemoryForComputation( in TEST_F()
Dbuffer_assignment.h598 void AssignBuffersFromHeapSimulator(const HeapSimulator::Result& result,
Dhlo.proto391 // A trace of a HeapSimulator run.