Class SoftmaxLayerLargeShapes

tests/datasets/ShapeDatasets.h:918–928

Data set containing large softmax layer shapes. */

/** Data set containing large softmax layer shapes. */
class SoftmaxLayerLargeShapes final : public ShapeDataset
{
public:
    SoftmaxLayerLargeShapes()
        : ShapeDataset("Shape",
                       {TensorShape{1000U, 10U}

                       })
    {
    }
};

/** Data set containing large and small softmax layer 4D shapes. */
class SoftmaxLayer4DShapes final : public ShapeDataset