Class LargeMatMulDataset

tests/datasets/LargeMatMulDataset.h:38–47

namespace datasets
{
class LargeMatMulDataset final : public MatMulDataset
{
public:
    LargeMatMulDataset()
    {
        add_config(TensorShape(21U, 13U, 3U, 2U), TensorShape(33U, 21U, 3U, 2U), TensorShape(33U, 13U, 3U, 2U));
        add_config(TensorShape(38U, 12U, 1U, 5U), TensorShape(21U, 38U, 1U, 5U), TensorShape(21U, 12U, 1U, 5U));
        add_config(TensorShape(45U, 38U, 3U, 2U), TensorShape(21U, 45U, 3U, 2U), TensorShape(21U, 38U, 3U, 2U));
    }
};

class HighDimensionalMatMulDataset final : public MatMulDataset
{