diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-09-01 18:06:04 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2021-09-15 11:23:05 +0000 |
commit | f37b970ff96b98310309e78aeea8a2e9df27b15a (patch) | |
tree | 73f1e3db6552649ac57901fd3df7fadc9742385e /src/armnn/Runtime.hpp | |
parent | fc8d434bb318aebb433a2f6d8ce9c066cd9c1b1e (diff) | |
download | armnn-f37b970ff96b98310309e78aeea8a2e9df27b15a.tar.gz |
IVGCVSW-6312 Support pre-importing inputs
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: Ifc5e6f2e36767cb2a5cbf281d40ec9989b581abc
Diffstat (limited to 'src/armnn/Runtime.hpp')
-rw-r--r-- | src/armnn/Runtime.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp index 7a80acd73e..e947dceb52 100644 --- a/src/armnn/Runtime.hpp +++ b/src/armnn/Runtime.hpp @@ -55,6 +55,8 @@ public: TensorInfo GetInputTensorInfo(NetworkId networkId, LayerBindingId layerId) const; TensorInfo GetOutputTensorInfo(NetworkId networkId, LayerBindingId layerId) const; + std::vector<ImportedInputId> ImportInputs(NetworkId networkId, const InputTensors& inputTensors); + // Evaluates network using input in inputTensors, outputs filled into outputTensors. Status EnqueueWorkload(NetworkId networkId, const InputTensors& inputTensors, @@ -66,7 +68,8 @@ public: /// Will block until this and any other thread using the same workingMem object completes. Status Execute(IWorkingMemHandle& workingMemHandle, const InputTensors& inputTensors, - const OutputTensors& outputTensors); + const OutputTensors& outputTensors, + std::vector<ImportedInputId> preImportedInputs); /// Unloads a network from the Runtime. /// At the moment this only removes the network from the m_Impl->m_Network. |