Class TinyShapes

tests/datasets/ShapeDatasets.h:146–158

Data set containing tiny tensor shapes. */

/** Data set containing tiny tensor shapes. */
class TinyShapes final : public ShapeDataset
{
public:
    TinyShapes()
        : ShapeDataset("Shape",
                       {
                           // Batch size 1
                           TensorShape{1U, 9U},
                           TensorShape{27U, 13U, 2U},
                       })
    {
    }
};
/** Data set containing small tensor shapes with none of the dimensions equal to 1 (unit). */
class SmallNoneUnitShapes final : public ShapeDataset
{