Class SmallDirectConvolutionShapes

tests/datasets/ShapeDatasets.h:756–769

Data set containing small tensor shapes for direct convolution. */

};
/** Data set containing small tensor shapes for direct convolution. */
class SmallDirectConvolutionShapes final : public ShapeDataset
{
public:
    SmallDirectConvolutionShapes()
        : ShapeDataset("InputShape",
                       {
                           // Batch size 1
                           TensorShape{32U, 37U, 3U},
                           // Batch size 4
                           TensorShape{6U, 9U, 5U, 4U},
                       })
    {
    }
};

class SmallDirectConv3DShapes final : public ShapeDataset
{