diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2022-06-22 15:20:14 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2022-06-24 11:29:03 +0000 |
commit | 3024650c5c2b5336aef438fa06a5d9bfafee2580 (patch) | |
tree | d8b9bbddf8616a0c940e529f731b83b21a415634 /shim/sl/canonical/ArmnnPreparedModel.hpp | |
parent | 09fa24d2f4b0177d55800bd01ec52c337701ef0a (diff) | |
download | armnn-3024650c5c2b5336aef438fa06a5d9bfafee2580.tar.gz |
IVGCVSW-6958 'Integrate Import Host Memory to POINTER memory type'
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I8b05fb1e6ba31e46bf23ebfdaf55142dbe17980e
Diffstat (limited to 'shim/sl/canonical/ArmnnPreparedModel.hpp')
-rw-r--r-- | shim/sl/canonical/ArmnnPreparedModel.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/shim/sl/canonical/ArmnnPreparedModel.hpp b/shim/sl/canonical/ArmnnPreparedModel.hpp index e97e7d7bd5..c487858c01 100644 --- a/shim/sl/canonical/ArmnnPreparedModel.hpp +++ b/shim/sl/canonical/ArmnnPreparedModel.hpp @@ -89,7 +89,8 @@ public: std::shared_ptr<std::vector<android::nn::RunTimePoolInfo>>& pMemPools, armnn::InputTensors& inputTensors, armnn::OutputTensors& outputTensors, - CanonicalExecutionContext callback) const; + CanonicalExecutionContext callback, + const bool pointerMemory = false) const; Priority GetModelPriority() const; @@ -113,7 +114,8 @@ private: ErrorStatus PrepareMemoryForIO(armnn::InputTensors& inputs, armnn::OutputTensors& outputs, std::vector<android::nn::RunTimePoolInfo>& memPools, - const Request& request) const; + const Request& request, + const bool pointerMemory = false) const; template <typename TensorBindingCollection> void DumpTensorsIfRequired(char const* tensorNamePrefix, const TensorBindingCollection& tensorBindings) const; |