aboutsummaryrefslogtreecommitdiff
path: root/shim/sl/canonical/ArmnnPreparedModel.hpp
diff options
context:
space:
mode:
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;