From f30f7d32b22020f80b21da7b008d8302cee9d395 Mon Sep 17 00:00:00 2001 From: Derek Lamberti Date: Tue, 9 Apr 2019 10:25:02 +0100 Subject: IVGCVSW-2946 RefElementwiseWorkload configures prior to first execute + Added PostAllocationConfigure() method to workload interface + Elementwise function now deduces types based on Functor - Replaced RefComparisonWorkload with RefElementwiseWorkload specialization + Fixed up unit tests and minor formatting Change-Id: I33d08797767bba01cf4efb2904920ce0f950a4fe Signed-off-by: Derek Lamberti --- src/armnn/LoadedNetwork.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/armnn') diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 9263f1a6e9..7f00dbee87 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -136,6 +136,12 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr net) // Set up memory. m_OptimizedNetwork->GetGraph().AllocateDynamicBuffers(); + + // Now that the intermediate tensor memory has been set-up, do any post allocation configuration for each workload. + for (auto& workload : m_WorkloadQueue) + { + workload->PostAllocationConfigure(); + } } TensorInfo LoadedNetwork::GetInputTensorInfo(LayerBindingId layerId) const -- cgit v1.2.1