Class Medium4DShapes

tests/datasets/ShapeDatasets.h:390–398

Data set containing medium 4D tensor shapes. */

/** Data set containing medium 4D tensor shapes. */
class Medium4DShapes final : public ShapeDataset
{
public:
    Medium4DShapes()
        : ShapeDataset("Shape",
                       {TensorShape{42U, 37U, 8U, 15U}, TensorShape{57U, 60U, 13U, 8U}, TensorShape{83U, 72U, 14U, 5U}})
    {
    }
};

/** Data set containing large tensor shapes. */
class LargeShapes final : public ShapeDataset