Searched refs:HloMemoryScheduler (Results 1 – 5 of 5) sorted by relevance
128 class HloMemoryScheduler : public HloModulePass {133 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,136 ~HloMemoryScheduler() override = default;
728 HloMemoryScheduler::HloMemoryScheduler( in HloMemoryScheduler() function in xla::HloMemoryScheduler733 StatusOr<bool> HloMemoryScheduler::Run(HloModule* module) { in Run()
90 HloMemoryScheduler scheduler([](const BufferValue& buffer) { in TEST_F()
299 HloMemoryScheduler scheduler(size_fn); in TEST_F()
47 HloMemoryScheduler scheduler( in RunHloRematerialization()