| 36 | namespace datasets |
| 37 | { |
| 38 | class SmallMatMulDataset final : public MatMulDataset |
| 39 | { |
| 40 | public: |
| 41 | SmallMatMulDataset() |
| 42 | { |
| 43 | add_config(TensorShape(3U, 4U, 2U, 2U), TensorShape(2U, 3U, 2U, 2U), TensorShape(2U, 4U, 2U, 2U)); |
| 44 | add_config(TensorShape(9U, 6U), TensorShape(5U, 9U), TensorShape(5U, 6U)); |
| 45 | add_config(TensorShape(31U, 1U), TensorShape(23U, 31U), TensorShape(23U, 1U)); |
| 46 | add_config(TensorShape(8U, 4U, 2U), TensorShape(16U, 8U, 2U), TensorShape(16U, 4U, 2U)); |
| 47 | add_config(TensorShape(32U, 2U), TensorShape(17U, 32U), TensorShape(17U, 2U)); |
| 48 | } |
| 49 | }; |
| 50 | |
| 51 | class SmallerMatMulDataset final : public MatMulDataset |
| 52 | { |
no outgoing calls
no test coverage detected