Lines Matching refs:Runner
94 SlotIndexes *const Indexes, MLModelRunner *Runner);
103 const MLModelRunner &getRunner() const { return *Runner; } in getRunner()
109 MLModelRunner *const Runner; member in llvm::MLPriorityAdvisor
142 if (!Runner) { in getAdvisor()
144 Runner = std::make_unique<ReleaseModeModelRunner<CompiledModelType>>( in getAdvisor()
147 Runner = std::make_unique<InteractiveModelRunner>( in getAdvisor()
153 MF, RA, &getAnalysis<SlotIndexes>(), Runner.get()); in getAdvisor()
155 std::unique_ptr<MLModelRunner> Runner; member in llvm::ReleaseModePriorityAdvisorAnalysis
180 MLModelRunner *Runner, Logger *Log) in DevelopmentModePriorityAdvisor() argument
181 : MLPriorityAdvisor(MF, RA, Indexes, Runner), Log(Log) {} in DevelopmentModePriorityAdvisor()
230 Runner = std::make_unique<NoInferenceModelRunner>(Ctx, InputFeatures); in doInitialization()
232 Runner = ModelUnderTrainingRunner::createAndEnsureValid( in doInitialization()
234 if (!Runner) { in doInitialization()
247 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(Runner.get())) in doInitialization()
261 if (!Runner) in getAdvisor()
268 MF, RA, &getAnalysis<SlotIndexes>(), Runner.get(), Log.get()); in getAdvisor()
271 std::unique_ptr<MLModelRunner> Runner; member in llvm::DevelopmentModePriorityAdvisorAnalysis
288 MLModelRunner *Runner) in MLPriorityAdvisor() argument
290 Runner(std::move(Runner)) { in MLPriorityAdvisor()
291 assert(this->Runner); in MLPriorityAdvisor()
292 Runner->switchContext(MF.getName()); in MLPriorityAdvisor()
299 *Runner->getTensor<int64_t>(0) = static_cast<int64_t>(Size); in getPriorityImpl()
300 *Runner->getTensor<int64_t>(1) = static_cast<int64_t>(Stage); in getPriorityImpl()
301 *Runner->getTensor<float>(2) = static_cast<float>(LI.weight()); in getPriorityImpl()
303 return Runner->evaluate<float>(); in getPriorityImpl()