Class SoftmaxLayerSmallShapes

tests/datasets/ShapeDatasets.h:902–915

Data set containing small softmax layer shapes. */

/** Data set containing small softmax layer shapes. */
class SoftmaxLayerSmallShapes final : public ShapeDataset
{
public:
    SoftmaxLayerSmallShapes()
        : ShapeDataset("Shape",
                       {
                           TensorShape{1U, 9U},
                           TensorShape{256U, 10U},
                           TensorShape{353U, 8U},
                           TensorShape{781U, 5U},
                       })
    {
    }
};

/** Data set containing large softmax layer shapes. */
class SoftmaxLayerLargeShapes final : public ShapeDataset