Class SMEMulShapesBroadcast

tests/datasets/ShapeDatasets.h:208–227

Data set containing pairs of small tensor shapes that are broadcast compatible. */

/** Data set containing pairs of small tensor shapes that are broadcast compatible. */
class SMEMulShapesBroadcast final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset>
{
public:
    SMEMulShapesBroadcast()
        : ZipDataset<ShapeDataset, ShapeDataset>(ShapeDataset("Shape0",
                                                              {
                                                                  // NOTE: This does not include x-wise broadcasting.
                                                                  TensorShape{9U, 9U},
                                                                  TensorShape{256U, 13U, 1U},
                                                                  TensorShape{128U, 1U, 5U, 1U},
                                                              }),
                                                 ShapeDataset("Shape1",
                                                              {
                                                                  TensorShape{9U, 1U, 2U},
                                                                  TensorShape{256U, 1U, 2U},
                                                                  TensorShape{128U, 64U, 1U, 3U},
                                                              }))
    {
    }
};

/** Data set containing small tensor shapes. */
class SmallShapesNoBatches final : public ShapeDataset