From 3024650c5c2b5336aef438fa06a5d9bfafee2580 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 22 Jun 2022 15:20:14 +0100 Subject: IVGCVSW-6958 'Integrate Import Host Memory to POINTER memory type' Signed-off-by: Sadik Armagan Change-Id: I8b05fb1e6ba31e46bf23ebfdaf55142dbe17980e --- shim/sl/canonical/ArmnnPreparedModel.hpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'shim/sl/canonical/ArmnnPreparedModel.hpp') 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>& 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& memPools, - const Request& request) const; + const Request& request, + const bool pointerMemory = false) const; template void DumpTensorsIfRequired(char const* tensorNamePrefix, const TensorBindingCollection& tensorBindings) const; -- cgit v1.2.1