diff options
Diffstat (limited to 'src/armnn/test/CreateWorkload.hpp')
-rw-r--r-- | src/armnn/test/CreateWorkload.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index b07197797c..47af4a89b5 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -41,11 +41,13 @@ std::unique_ptr<Workload> MakeAndCheckWorkload(Layer& layer, Graph& graph, const } // Helper function to create tensor handlers for workloads, assuming they all use the same factory. -void CreateTensorHandles(armnn::Graph& graph, armnn::IWorkloadFactory& factory) +void CreateTensorHandles(armnn::Graph& graph, + armnn::IWorkloadFactory& factory) { + TensorHandleFactoryRegistry tmpRegistry; for (auto&& layer : graph.TopologicalSort()) { - layer->CreateTensorHandles(graph, factory); + layer->CreateTensorHandles(tmpRegistry, factory); } } |