Class TinyShapesBroadcast

tests/datasets/ShapeDatasets.h:243–259

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

/** Data set containing pairs of tiny tensor shapes that are broadcast compatible. */
class TinyShapesBroadcast final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset>
{
public:
    TinyShapesBroadcast()
        : ZipDataset<ShapeDataset, ShapeDataset>(ShapeDataset("Shape0",
                                                              {
                                                                  TensorShape{9U, 9U},
                                                                  TensorShape{10U, 2U, 14U, 2U},
                                                              }),
                                                 ShapeDataset("Shape1",
                                                              {
                                                                  TensorShape{9U, 1U, 9U},
                                                                  TensorShape{10U},
                                                              }))
    {
    }
};
/** Data set containing pairs of tiny tensor shapes that are broadcast compatible and can do in_place calculation. */
class TinyShapesBroadcastInplace final : public framework::dataset::ZipDataset<ShapeDataset, ShapeDataset>
{