diff options
author | Finn Williams <finwil01@e127804.cambridge.arm.com> | 2021-10-02 15:06:39 +0100 |
---|---|---|
committer | Finn Williams <finwil01@e127804.cambridge.arm.com> | 2021-10-08 15:42:14 +0100 |
commit | 8636bc705cc33fd869f64ebf24b14836d5a40b29 (patch) | |
tree | e546319af87596d053c75d65eedd1efb1f946228 /src/armnn/Runtime.hpp | |
parent | 521032fd424cf86681eb125afbf5eaee47d8c585 (diff) | |
download | armnn-8636bc705cc33fd869f64ebf24b14836d5a40b29.tar.gz |
IVGCVSW-6313 Support pre-importing outputs
* Add ClearImportedInputs/Outputs function to IRuntime
* Add UnImport function to ITensorHandle
* Remove mutex from IWorkingMemHandle
Change-Id: I34c9b6e1618755e10f3b4597afa1d9a9ea97e5fe
Signed-off-by: Finn Williams <finwil01@e127804.cambridge.arm.com>
Diffstat (limited to 'src/armnn/Runtime.hpp')
-rw-r--r-- | src/armnn/Runtime.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp index e947dceb52..05de372eee 100644 --- a/src/armnn/Runtime.hpp +++ b/src/armnn/Runtime.hpp @@ -56,6 +56,10 @@ public: TensorInfo GetOutputTensorInfo(NetworkId networkId, LayerBindingId layerId) const; std::vector<ImportedInputId> ImportInputs(NetworkId networkId, const InputTensors& inputTensors); + std::vector<ImportedOutputId> ImportOutputs(NetworkId networkId, const OutputTensors& outputTensors); + + void ClearImportedInputs(NetworkId networkId, const std::vector<ImportedInputId> inputIds); + void ClearImportedOutputs(NetworkId networkId, const std::vector<ImportedOutputId> outputIds); // Evaluates network using input in inputTensors, outputs filled into outputTensors. Status EnqueueWorkload(NetworkId networkId, @@ -69,7 +73,8 @@ public: Status Execute(IWorkingMemHandle& workingMemHandle, const InputTensors& inputTensors, const OutputTensors& outputTensors, - std::vector<ImportedInputId> preImportedInputs); + std::vector<ImportedInputId> preImportedInputs, + std::vector<ImportedOutputId> preImportedOutputs); /// Unloads a network from the Runtime. /// At the moment this only removes the network from the m_Impl->m_Network. |