diff options
Diffstat (limited to 'tests/datasets/SmallMatMulDataset.h')
-rw-r--r-- | tests/datasets/SmallMatMulDataset.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tests/datasets/SmallMatMulDataset.h b/tests/datasets/SmallMatMulDataset.h index 52ef01da7b..bb4cdad54b 100644 --- a/tests/datasets/SmallMatMulDataset.h +++ b/tests/datasets/SmallMatMulDataset.h @@ -47,6 +47,17 @@ public: } }; +class SmallerMatMulDataset final : public MatMulDataset +{ +public: + SmallerMatMulDataset() + { + add_config(TensorShape(9U, 6U), TensorShape(5U, 9U), TensorShape(5U, 6U)); + add_config(TensorShape(8U, 4U, 2U), TensorShape(16U, 8U, 2U), TensorShape(16U, 4U, 2U)); + add_config(TensorShape(32U, 2U), TensorShape(17U, 32U), TensorShape(17U, 2U)); + } +}; + class TinyMatMulDataset final : public MatMulDataset { public: |