Data set containing large 4D tensor shapes. */
| 468 | |
| 469 | /** Data set containing large 4D tensor shapes. */ |
| 470 | class Large4DShapes final : public ShapeDataset |
| 471 | { |
| 472 | public: |
| 473 | Large4DShapes() : ShapeDataset("Shape", {TensorShape{320U, 123U, 3U, 3U}}) |
| 474 | { |
| 475 | } |
| 476 | }; |
| 477 | |
| 478 | /** Data set containing small 3x3 tensor shapes. */ |
| 479 | class Small3x3Shapes final : public ShapeDataset |
no outgoing calls
no test coverage detected