Lines Matching refs:Runner

221 void resetInputs(MLModelRunner &Runner) {  in resetInputs()  argument
223 std::memset(Runner.getTensorUntyped(FeatureIDs::NAME), 0, \ in resetInputs()
250 MLModelRunner *Runner, const MachineBlockFrequencyInfo &MBFI,
260 const MLModelRunner &getRunner() const { return *Runner; } in getRunner()
309 MLModelRunner *const Runner; member in __anon8d2f77590111::MLEvictAdvisor
351 if (!Runner) in getAdvisor()
352 Runner = std::make_unique<ReleaseModeModelRunner<CompiledModelType>>( in getAdvisor()
355 MF, RA, Runner.get(), getAnalysis<MachineBlockFrequencyInfo>(), in getAdvisor()
358 std::unique_ptr<ReleaseModeModelRunner<CompiledModelType>> Runner; member in __anon8d2f77590111::ReleaseModeEvictionAdvisorAnalysis
388 MLModelRunner *Runner, in DevelopmentModeEvictAdvisor() argument
391 : MLEvictAdvisor(MF, RA, Runner, MBFI, Loops), Log(Log) {} in DevelopmentModeEvictAdvisor()
450 if (!Runner) { in getAdvisor()
452 Runner = std::make_unique<NoInferenceModelRunner>(Ctx, InputFeatures); in getAdvisor()
454 Runner = ModelUnderTrainingRunner::createAndEnsureValid( in getAdvisor()
456 if (!Runner) { in getAdvisor()
467 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(Runner.get())) in getAdvisor()
481 MF, RA, Runner.get(), getAnalysis<MachineBlockFrequencyInfo>(), in getAdvisor()
485 std::unique_ptr<MLModelRunner> Runner; member in __anon8d2f77590111::DevelopmentModeEvictionAdvisorAnalysis
504 MLModelRunner *Runner, in MLEvictAdvisor() argument
508 Runner(std::move(Runner)), MBFI(MBFI), Loops(Loops), in MLEvictAdvisor()
510 assert(this->Runner); in MLEvictAdvisor()
523 int64_t Ret = Runner->evaluate<int64_t>(); in tryFindEvictionCandidatePosition()
615 resetInputs(*Runner); in tryFindEvictionCandidate()
674 Runner->getTensor<float>(FeatureIndex)[Pos] /= Largest[FeatureIndex]; in tryFindEvictionCandidate()
677 *Runner->getTensor<float>(FeatureIDs::progress) = in tryFindEvictionCandidate()
811 Runner->getTensor<TYPE>(FeatureIDs::ID)[Pos] = static_cast<TYPE>(VAL); \ in extractFeatures()