diff options
Diffstat (limited to 'include/armnn/backends/WorkloadData.hpp')
| -rw-r--r-- | include/armnn/backends/WorkloadData.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/include/armnn/backends/WorkloadData.hpp b/include/armnn/backends/WorkloadData.hpp index 1a2f34e21f..00962ed52c 100644 --- a/include/armnn/backends/WorkloadData.hpp +++ b/include/armnn/backends/WorkloadData.hpp @@ -785,4 +785,9 @@ struct ChannelShuffleQueueDescriptor : QueueDescriptorWithParameters<ChannelShuf void Validate(const WorkloadInfo& workloadInfo) const; }; +struct BatchMatMulQueueDescriptor : QueueDescriptorWithParameters<BatchMatMulDescriptor> +{ + void Validate(const WorkloadInfo& workloadInfo) const; +}; + } // namespace armnn |