Class GlobalPoolingShapes

tests/datasets/ShapeDatasets.h:875–886

Data set containing global pooling tensor shapes. */

/** Data set containing global pooling tensor shapes. */
class GlobalPoolingShapes final : public ShapeDataset
{
public:
    GlobalPoolingShapes()
        : ShapeDataset("Shape",
                       {// Batch size 1
                        TensorShape{9U, 9U}, TensorShape{13U, 13U, 2U}, TensorShape{27U, 27U, 1U, 3U},
                        // Batch size 4
                        TensorShape{31U, 31U, 3U, 4U}, TensorShape{34U, 34U, 2U, 4U}})
    {
    }
};
/** Data set containing tiny softmax layer shapes. */
class SoftmaxLayerTinyShapes final : public ShapeDataset
{