Searched refs:HeapSimulator (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | heap_simulator.cc | 33 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 …]
|
D | heap_simulator.h | 47 class HeapSimulator { 140 HeapSimulator(std::unique_ptr<HeapAlgorithm> algorithm, 145 ~HeapSimulator(); 213 using Chunk = HeapSimulator::Chunk; 214 using Result = HeapSimulator::Result;
|
D | heap_simulator_test.cc | 98 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 …]
|
D | hlo_memory_scheduler.cc | 528 HeapSimulator::MinimumMemoryForComputation( in DefaultMemoryScheduler() 537 HeapSimulator::MinimumMemoryForComputation( in DefaultMemoryScheduler() 547 HeapSimulator::MinimumMemoryForComputation( in DefaultMemoryScheduler() 584 HeapSimulator::MinimumMemoryForComputation( in ScheduleModule()
|
D | buffer_assignment.cc | 649 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()
|
D | hlo_memory_scheduler_test.cc | 302 EXPECT_EQ(16, HeapSimulator::MinimumMemoryForComputation( in TEST_F() 308 EXPECT_EQ(17, HeapSimulator::MinimumMemoryForComputation( in TEST_F()
|
D | buffer_assignment.h | 598 void AssignBuffersFromHeapSimulator(const HeapSimulator::Result& result,
|
D | hlo.proto | 391 // A trace of a HeapSimulator run.
|