diff options
Diffstat (limited to 'src/backends/reference/workloads/RefBatchNormalizationWorkload.cpp')
| -rw-r--r-- | src/backends/reference/workloads/RefBatchNormalizationWorkload.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/RefBatchNormalizationWorkload.cpp b/src/backends/reference/workloads/RefBatchNormalizationWorkload.cpp index 282374d89b..a6bd986f1d 100644 --- a/src/backends/reference/workloads/RefBatchNormalizationWorkload.cpp +++ b/src/backends/reference/workloads/RefBatchNormalizationWorkload.cpp @@ -15,7 +15,7 @@ namespace armnn RefBatchNormalizationWorkload::RefBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) - : BaseWorkload(descriptor, info) + : RefBaseWorkload(descriptor, info) , m_Mean (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Mean))) , m_Variance(std::make_unique<ScopedTensorHandle>(*(descriptor.m_Variance))) , m_Beta (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Beta))) |