Class Small4DShapes

tests/datasets/ShapeDatasets.h:134–143

Data set containing small 4D tensor shapes. */

};
/** Data set containing small 4D tensor shapes. */
class Small4DShapes final : public ShapeDataset
{
public:
    Small4DShapes()
        : ShapeDataset("Shape",
                       {TensorShape{2U, 7U, 1U, 3U}, TensorShape{7U, 7U, 5U, 3U}, TensorShape{27U, 13U, 37U, 2U},
                        TensorShape{128U, 64U, 21U, 3U}})
    {
    }
};

/** Data set containing tiny tensor shapes. */
class TinyShapes final : public ShapeDataset