Class SmallNoneUnitShapes

tests/datasets/ShapeDatasets.h:160–174

Data set containing small tensor shapes with none of the dimensions equal to 1 (unit). */

};
/** Data set containing small tensor shapes with none of the dimensions equal to 1 (unit). */
class SmallNoneUnitShapes final : public ShapeDataset
{
public:
    SmallNoneUnitShapes()
        : ShapeDataset("Shape",
                       {// Batch size 1
                        TensorShape{13U, 11U}, TensorShape{16U, 16U}, TensorShape{24U, 26U, 5U},
                        TensorShape{7U, 7U, 17U, 2U},
                        // Batch size 4
                        TensorShape{27U, 13U, 2U, 4U},
                        // Arbitrary batch size
                        TensorShape{8U, 7U, 5U, 5U}})
    {
    }
};
/** Data set containing small tensor shapes. */
class SmallShapes final : public ShapeDataset
{