Lines Matching refs:Runner
95 SlotIndexes *const Indexes, MLModelRunner *Runner);
104 const MLModelRunner &getRunner() const { return *Runner; } in getRunner()
110 MLModelRunner *const Runner; member in llvm::MLPriorityAdvisor
143 if (!Runner) { in getAdvisor()
145 Runner = std::make_unique<ReleaseModeModelRunner<CompiledModelType>>( in getAdvisor()
148 Runner = std::make_unique<InteractiveModelRunner>( in getAdvisor()
154 MF, RA, &getAnalysis<SlotIndexesWrapperPass>().getSI(), Runner.get()); in getAdvisor()
156 std::unique_ptr<MLModelRunner> Runner; member in llvm::ReleaseModePriorityAdvisorAnalysis
181 MLModelRunner *Runner, Logger *Log) in DevelopmentModePriorityAdvisor() argument
182 : MLPriorityAdvisor(MF, RA, Indexes, Runner), Log(Log) {} in DevelopmentModePriorityAdvisor()
231 Runner = std::make_unique<NoInferenceModelRunner>(Ctx, InputFeatures); in doInitialization()
233 Runner = ModelUnderTrainingRunner::createAndEnsureValid( in doInitialization()
235 if (!Runner) { in doInitialization()
248 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(Runner.get())) in doInitialization()
262 if (!Runner) in getAdvisor()
269 MF, RA, &getAnalysis<SlotIndexesWrapperPass>().getSI(), Runner.get(), in getAdvisor()
273 std::unique_ptr<MLModelRunner> Runner; member in llvm::DevelopmentModePriorityAdvisorAnalysis
290 MLModelRunner *Runner) in MLPriorityAdvisor() argument
292 Runner(std::move(Runner)) { in MLPriorityAdvisor()
293 assert(this->Runner); in MLPriorityAdvisor()
294 Runner->switchContext(MF.getName()); in MLPriorityAdvisor()
301 *Runner->getTensor<int64_t>(0) = static_cast<int64_t>(Size); in getPriorityImpl()
302 *Runner->getTensor<int64_t>(1) = static_cast<int64_t>(Stage); in getPriorityImpl()
303 *Runner->getTensor<float>(2) = static_cast<float>(LI.weight()); in getPriorityImpl()
305 return Runner->evaluate<float>(); in getPriorityImpl()