Searched refs:HloMemoryScheduler (Results 1 – 5 of 5) sorted by relevance
94 class HloMemoryScheduler : public HloModulePass {99 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,101 ~HloMemoryScheduler() override = default;
609 HloMemoryScheduler::HloMemoryScheduler( in HloMemoryScheduler() function in xla::HloMemoryScheduler614 StatusOr<bool> HloMemoryScheduler::Run(HloModule* module) { in Run()
301 HloMemoryScheduler scheduler(size_fn); in TEST_F()
71 HloMemoryScheduler scheduler([](const BufferValue& buffer) { in TEST_F()
147 HloMemoryScheduler scheduler( in RunHloRematerialization()