Data set containing small tensor shapes. */
| 174 | }; |
| 175 | /** Data set containing small tensor shapes. */ |
| 176 | class SmallShapes final : public ShapeDataset |
| 177 | { |
| 178 | public: |
| 179 | SmallShapes() |
| 180 | : ShapeDataset("Shape", |
| 181 | {// Batch size 1 |
| 182 | //TensorShape{ 3U, 11U }, |
| 183 | //TensorShape{ 1U, 16U }, |
| 184 | TensorShape{27U, 13U, 7U}, TensorShape{7U, 7U, 17U, 2U}, |
| 185 | // Batch size 4 and 2 SIMD iterations |
| 186 | TensorShape{33U, 13U, 2U, 4U}, |
| 187 | // Arbitrary batch size |
| 188 | TensorShape{11U, 11U, 3U, 5U}}) |
| 189 | { |
| 190 | } |
| 191 | }; |
| 192 | |
| 193 | class SMEMulShapes final : public ShapeDataset |
| 194 | { |
no outgoing calls
no test coverage detected