Home
last modified time | relevance | path

Searched refs:MLModelRunner (Results 1 – 13 of 13) sorted by relevance

/src/contrib/llvm-project/llvm/include/llvm/Analysis/
H A DMLModelRunner.h26 class MLModelRunner {
29 MLModelRunner(const MLModelRunner &) = delete;
30 MLModelRunner &operator=(const MLModelRunner &) = delete;
31 virtual ~MLModelRunner() = default;
49 return (const_cast<MLModelRunner *>(this))->getTensorUntyped(Index); in getTensorUntyped()
57 MLModelRunner(LLVMContext &Ctx, Kind Type, size_t NrInputs) in MLModelRunner() function
H A DReleaseModeModelRunner.h55 class ReleaseModeModelRunner final : public MLModelRunner {
63 : MLModelRunner(Ctx, MLModelRunner::Kind::Release, InputSpec.size() + 1),
111 static bool classof(const MLModelRunner *R) { in classof()
112 return R->getKind() == MLModelRunner::Kind::Release; in classof()
H A DNoInferenceModelRunner.h20 class NoInferenceModelRunner : public MLModelRunner {
25 static bool classof(const MLModelRunner *R) { in classof()
26 return R->getKind() == MLModelRunner::Kind::NoOp; in classof()
H A DInteractiveModelRunner.h41 class InteractiveModelRunner : public MLModelRunner {
48 static bool classof(const MLModelRunner *R) { in classof()
49 return R->getKind() == MLModelRunner::Kind::Interactive; in classof()
H A DModelUnderTrainingRunner.h31 class ModelUnderTrainingRunner final : public MLModelRunner {
50 static bool classof(const MLModelRunner *R) { in classof()
51 return R->getKind() == MLModelRunner::Kind::Development; in classof()
H A DMLInlineAdvisor.h32 std::unique_ptr<MLModelRunner> ModelRunner,
48 const MLModelRunner &getModelRunner() const { return *ModelRunner; } in getModelRunner()
67 std::unique_ptr<MLModelRunner> ModelRunner;
/src/contrib/llvm-project/llvm/lib/CodeGen/
H A DMLRegAllocPriorityAdvisor.cpp95 SlotIndexes *const Indexes, MLModelRunner *Runner);
104 const MLModelRunner &getRunner() const { return *Runner; } in getRunner()
110 MLModelRunner *const Runner;
156 std::unique_ptr<MLModelRunner> Runner;
181 MLModelRunner *Runner, Logger *Log) in DevelopmentModePriorityAdvisor()
273 std::unique_ptr<MLModelRunner> Runner;
290 MLModelRunner *Runner) in MLPriorityAdvisor()
H A DMLRegAllocEvictAdvisor.h37 MLModelRunner *RegallocRunner, function_ref<int(SlotIndex)> GetOpcode,
49 MLModelRunner *RegallocRunner, const int MBBFreqIndex,
H A DMLRegAllocEvictAdvisor.cpp256 void resetInputs(MLModelRunner &Runner) { in resetInputs()
290 MLModelRunner *Runner, const MachineBlockFrequencyInfo &MBFI,
300 const MLModelRunner &getRunner() const { return *Runner; } in getRunner()
350 MLModelRunner *const Runner;
413 std::unique_ptr<MLModelRunner> Runner;
434 MLModelRunner *Runner, in DevelopmentModeEvictAdvisor()
553 std::unique_ptr<MLModelRunner> Runner;
573 MLModelRunner *Runner, in MLEvictAdvisor()
947 SmallVectorImpl<LRStartEndInfo> &LRPosInfo, MLModelRunner *RegallocRunner, in extractInstructionFeatures()
1068 MLModelRunner *RegallocRunner, const int MBBFreqIndex, in extractMBBFrequency()
/src/contrib/llvm-project/llvm/lib/Analysis/
H A DNoInferenceModelRunner.cpp19 : MLModelRunner(Ctx, MLModelRunner::Kind::NoOp, Inputs.size()) { in NoInferenceModelRunner()
H A DInteractiveModelRunner.cpp29 : MLModelRunner(Ctx, MLModelRunner::Kind::Interactive, Inputs.size()), in InteractiveModelRunner()
H A DModelUnderTrainingRunner.cpp98 : MLModelRunner(Ctx, MLModelRunner::Kind::Development, InputSpecs.size()), in ModelUnderTrainingRunner()
H A DMLInlineAdvisor.cpp77 std::unique_ptr<MLModelRunner> AOTRunner; in getReleaseModeAdvisor()
141 std::unique_ptr<MLModelRunner> Runner, in MLInlineAdvisor()