Class SmallScatterMultiDimDataset

tests/datasets/ScatterDataset.h:130–144

This dataset represents the (m+1)-D updates/dst case.

// This dataset represents the (m+1)-D updates/dst case.
class SmallScatterMultiDimDataset final : public ScatterDataset
{
public:
    SmallScatterMultiDimDataset()
    {
        // NOTE: Config is src, updates, indices, output.
        //      - In this config, the dim replaced is the final number (largest tensor dimension)
        //      - Largest "updates" dim should match y-dim of indices.
        //      - src/updates/dst should all have same number of dims. Indices should be 2D.
        add_config(TensorShape(6U, 5U), TensorShape(6U, 2U), TensorShape(1U, 2U), TensorShape(6U, 5U));
        add_config(TensorShape(9U, 3U, 4U), TensorShape(9U, 3U, 2U), TensorShape(1U, 2U), TensorShape(9U, 3U, 4U));
        add_config(TensorShape(17U, 3U, 2U, 4U), TensorShape(17U, 3U, 2U, 7U), TensorShape(1U, 7U),
                   TensorShape(17U, 3U, 2U, 4U));
    }
};

// This dataset represents the (m+1)-D updates tensor, (m+n)-d output tensor cases
class SmallScatterMultiIndicesDataset final : public ScatterDataset