Class Small2DNonMaxSuppressionShapes

tests/datasets/ShapeDatasets.h:1040–1047

Data set containing small 2D tensor shapes. */

/** Data set containing small 2D tensor shapes. */
class Small2DNonMaxSuppressionShapes final : public ShapeDataset
{
public:
    Small2DNonMaxSuppressionShapes()
        : ShapeDataset("Shape", {TensorShape{4U, 7U}, TensorShape{4U, 13U}, TensorShape{4U, 64U}})
    {
    }
};

/** Data set containing large 2D tensor shapes. */
class Large2DNonMaxSuppressionShapes final : public ShapeDataset