Class Small1DShapes

tests/datasets/ShapeDatasets.h:57–64

Data set containing small 1D tensor shapes. */

/** Data set containing small 1D tensor shapes. */
class Small1DShapes final : public ShapeDataset
{
public:
    Small1DShapes()
        : ShapeDataset("Shape", {TensorShape{128U}, TensorShape{256U}, TensorShape{512U}, TensorShape{1024U}})
    {
    }
};

/** Data set containing tiny 2D tensor shapes. */
class Tiny2DShapes final : public ShapeDataset