diff options
Diffstat (limited to 'ArmnnPreparedModel_1_2.hpp')
-rw-r--r-- | ArmnnPreparedModel_1_2.hpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/ArmnnPreparedModel_1_2.hpp b/ArmnnPreparedModel_1_2.hpp index c64c891e..255fc187 100644 --- a/ArmnnPreparedModel_1_2.hpp +++ b/ArmnnPreparedModel_1_2.hpp @@ -49,6 +49,14 @@ public: const bool asyncModelExecutionEnabled = false, const unsigned int numberOfThreads = 1); + ArmnnPreparedModel_1_2(armnn::NetworkId networkId, + armnn::IRuntime* runtime, + const std::string& requestInputsAndOutputsDumpDir, + const bool gpuProfilingEnabled, + const bool asyncModelExecutionEnabled = false, + const unsigned int numberOfThreads = 1, + const bool preparedFromCache = false); + virtual ~ArmnnPreparedModel_1_2(); virtual Return<V1_0::ErrorStatus> execute(const V1_0::Request& request, @@ -76,7 +84,7 @@ public: /// Executes this model with dummy inputs (e.g. all zeroes). /// \return false on failure, otherwise true - bool ExecuteWithDummyInputs(); + bool ExecuteWithDummyInputs(unsigned int numInputs, unsigned int numOutputs); private: @@ -156,6 +164,7 @@ private: static std::unique_ptr<armnn::Threadpool> m_Threadpool; std::shared_ptr<IWorkingMemHandle> m_WorkingMemHandle; const bool m_AsyncModelExecutionEnabled; + const bool m_PreparedFromCache; }; } |