Class SmallScatterScalarDataset

tests/datasets/ScatterDataset.h:213–224

};

class SmallScatterScalarDataset final : public ScatterDataset
{
public:
    // batched scalar case
    SmallScatterScalarDataset()
    {
        add_config(TensorShape(6U, 5U), TensorShape(6U), TensorShape(2U, 6U), TensorShape(6U, 5U));
        add_config(TensorShape(6U, 5U), TensorShape(6U, 6U), TensorShape(2U, 6U, 6U), TensorShape(6U, 5U));
        add_config(TensorShape(3U, 3U, 6U, 5U), TensorShape(6U, 6U), TensorShape(4U, 6U, 6U),
                   TensorShape(3U, 3U, 6U, 5U));
    }
};

// This dataset is for data types that does not require full testing. It contains selected tests from the above.
class SmallScatterMixedDataset final : public ScatterDataset