Class SMEMulShapes

tests/datasets/ShapeDatasets.h:193–205

};

class SMEMulShapes final : public ShapeDataset
{
public:
    SMEMulShapes()
        : ShapeDataset(
              "Shape",
              {
                  // Batch size 1
                  TensorShape{6U, 1U}, TensorShape{128U, 2U}, TensorShape{128U, 2U, 4U, 2U, 2U} // 5D collapsible case
              })
    {
    }
};

/** Data set containing pairs of small tensor shapes that are broadcast compatible. */
class SMEMulShapesBroadcast final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset>