aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/WorkloadData.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/WorkloadData.hpp')
-rw-r--r--src/armnn/backends/WorkloadData.hpp11
1 files changed, 1 insertions, 10 deletions
diff --git a/src/armnn/backends/WorkloadData.hpp b/src/armnn/backends/WorkloadData.hpp
index face761e73..a36f0ad7e6 100644
--- a/src/armnn/backends/WorkloadData.hpp
+++ b/src/armnn/backends/WorkloadData.hpp
@@ -197,17 +197,8 @@ struct SubtractionQueueDescriptor : QueueDescriptor
};
// Mean layer workload data.
-struct MeanQueueDescriptor : QueueDescriptor
+struct MeanQueueDescriptor : QueueDescriptorWithParameters<MeanDescriptor>
{
- MeanQueueDescriptor()
- : m_Axis(nullptr)
- , m_Keepdims(false)
- {
- }
-
- const ConstCpuTensorHandle* m_Axis;
- bool m_Keepdims;
-
void Validate(const WorkloadInfo& workloadInfo) const;
};