Class SmallGEMMReshape2DShapes

tests/datasets/ShapeDatasets.h:996–1006

Data set containing small tensor shapes to be used with the GEMM reshaping kernel */

/** Data set containing small tensor shapes to be used with the GEMM reshaping kernel */
class SmallGEMMReshape2DShapes final : public ShapeDataset
{
public:
    SmallGEMMReshape2DShapes()
        : ShapeDataset("Shape",
                       {
                           TensorShape{63U, 72U},
                       })
    {
    }
};

/** Data set containing small tensor shapes to be used with the GEMM reshaping kernel when the input has to be reinterpreted as 3D */
class SmallGEMMReshape3DShapes final : public ShapeDataset