Searched refs:ModelUnderTrainingRunner (Results 1 – 6 of 6) sorted by relevance
1 //===- ModelUnderTrainingRunner.h -- 'development' mode runner --*- C++ -*-===//26 /// ModelUnderTrainingRunner - training mode implementation. It uses TFLite31 class ModelUnderTrainingRunner final : public MLModelRunner {34 ModelUnderTrainingRunner(const ModelUnderTrainingRunner &) = delete;35 ModelUnderTrainingRunner &36 operator=(const ModelUnderTrainingRunner &) = delete;54 static std::unique_ptr<ModelUnderTrainingRunner>60 ModelUnderTrainingRunner(
93 ModelUnderTrainingRunner::ModelUnderTrainingRunner( in ModelUnderTrainingRunner() function in ModelUnderTrainingRunner113 void *ModelUnderTrainingRunner::evaluateUntyped() { in evaluateUntyped()122 std::unique_ptr<ModelUnderTrainingRunner>123 ModelUnderTrainingRunner::createAndEnsureValid( in createAndEnsureValid()129 std::unique_ptr<ModelUnderTrainingRunner> MUTR; in createAndEnsureValid()145 MUTR.reset(new ModelUnderTrainingRunner( in createAndEnsureValid()
233 Runner = ModelUnderTrainingRunner::createAndEnsureValid( in doInitialization()248 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(Runner.get())) in doInitialization()321 if (isa<ModelUnderTrainingRunner>(getRunner())) { in getPriority()344 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(&getRunner())) { in getPriority()
514 Runner = ModelUnderTrainingRunner::createAndEnsureValid( in doInitialization()529 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(Runner.get())) in doInitialization()1092 if (isa<ModelUnderTrainingRunner>(getRunner())) { in tryFindEvictionCandidatePosition()1127 if (auto *MUTR = dyn_cast<ModelUnderTrainingRunner>(&getRunner())) in tryFindEvictionCandidatePosition()