From 840c45d19bff23d64f78a7e466886fb970b4fcc9 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Tue, 15 Feb 2022 20:47:34 +0000 Subject: Refactor Forced Import * Find and replace all workloads associated with imported IO * Only attempt tensorhandle replacement if supported by all workloads * Add new RefBaseWorkload to enable forced input for ref backend * Store imported tensorhandles in preImportedTensorhandles instead of outputHandles * Create pre-imported tensorhandles at network load-time * Front load import workload validation to load network time * Only call ReplaceTensorHandle when needed Change-Id: I3816a71b7f57ae90388bb16462a75d4ef3544fa7 Signed-off-by: Finn Williams --- src/backends/reference/workloads/RefDequantizeWorkload.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/backends/reference/workloads/RefDequantizeWorkload.hpp') diff --git a/src/backends/reference/workloads/RefDequantizeWorkload.hpp b/src/backends/reference/workloads/RefDequantizeWorkload.hpp index 285c6496bb..8fa8951677 100644 --- a/src/backends/reference/workloads/RefDequantizeWorkload.hpp +++ b/src/backends/reference/workloads/RefDequantizeWorkload.hpp @@ -5,16 +5,16 @@ #pragma once -#include +#include "RefBaseWorkload.hpp" namespace armnn { -class RefDequantizeWorkload : public BaseWorkload +class RefDequantizeWorkload : public RefBaseWorkload { public: - using BaseWorkload::m_Data; - using BaseWorkload::BaseWorkload; + using RefBaseWorkload::m_Data; + using RefBaseWorkload::RefBaseWorkload; void Execute() const override; void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; -- cgit v1.2.1