Searched refs:set_schedule (Results 1 – 5 of 5) sorted by relevance
44 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule() function in xla::HloModule362 TF_RETURN_IF_ERROR(module->set_schedule(std::move(schedule))); in CreateFromProto()586 TF_CHECK_OK(module->set_schedule(std::move(clone_schedule))); in Clone()
262 Status set_schedule(HloSchedule schedule);
617 TF_RETURN_IF_ERROR(module->set_schedule(std::move(schedule))); in Run()634 TF_RETURN_IF_ERROR(module->set_schedule(std::move(schedule))); in Run()
504 TF_CHECK_OK(module->set_schedule(ScheduleFromInstructionOrder(module))); in ParseHloModule()
272 void set_schedule(Schedule* schedule) { in set_schedule() function in v8::internal::compiler::PipelineData1648 data->set_schedule(schedule); in Run()