diff options
Diffstat (limited to 'src/armnn/backends/WorkloadData.hpp')
| -rw-r--r-- | src/armnn/backends/WorkloadData.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/backends/WorkloadData.hpp b/src/armnn/backends/WorkloadData.hpp index d50a237273..f8f7e326dc 100644 --- a/src/armnn/backends/WorkloadData.hpp +++ b/src/armnn/backends/WorkloadData.hpp @@ -196,6 +196,12 @@ struct SubtractionQueueDescriptor : QueueDescriptor void Validate(const WorkloadInfo& workloadInfo) const; }; +// Mean layer workload data. +struct MeanQueueDescriptor : QueueDescriptor +{ + void Validate(const WorkloadInfo& workloadInfo) const; +}; + // Batch norm layer workload data. struct BatchNormalizationQueueDescriptor : QueueDescriptorWithParameters<BatchNormalizationDescriptor> { |