aboutsummaryrefslogtreecommitdiff
path: root/shim/sl/canonical/ArmnnPreparedModel.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2022-06-22 15:20:14 +0100
committerSadik Armagan <sadik.armagan@arm.com>2022-06-24 11:29:03 +0000
commit3024650c5c2b5336aef438fa06a5d9bfafee2580 (patch)
treed8b9bbddf8616a0c940e529f731b83b21a415634 /shim/sl/canonical/ArmnnPreparedModel.hpp
parent09fa24d2f4b0177d55800bd01ec52c337701ef0a (diff)
downloadarmnn-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.hpp6
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;