Class SmallScatterMultiIndicesDataset

tests/datasets/ScatterDataset.h:147–181

This dataset represents the (m+1)-D updates tensor, (m+n)-d output tensor cases

// This dataset represents the (m+1)-D updates tensor, (m+n)-d output tensor cases
class SmallScatterMultiIndicesDataset final : public ScatterDataset
{
public:
    SmallScatterMultiIndicesDataset()
    {
        // NOTE: Config is src, updates, indices, output.
        // NOTE: indices.shape.x = src.num_dimensions - updates.num_dimensions + 1

        // index length is 2
        add_config(TensorShape(6U, 5U, 2U), TensorShape(6U, 4U), TensorShape(2U, 4U), TensorShape(6U, 5U, 2U));
        add_config(TensorShape(17U, 3U, 3U, 2U), TensorShape(17U, 3U, 2U), TensorShape(2U, 2U),
                   TensorShape(17U, 3U, 3U, 2U));
        add_config(TensorShape(11U, 3U, 3U, 2U, 4U), TensorShape(11U, 3U, 3U, 4U), TensorShape(2U, 4U),
                   TensorShape(11U, 3U, 3U, 2U, 4U));
        add_config(TensorShape(5U, 4U, 3U, 3U, 2U, 4U), TensorShape(5U, 4U, 3U, 3U, 5U), TensorShape(2U, 5U),
                   TensorShape(5U, 4U, 3U, 3U, 2U, 4U));

        // index length is 3
        add_config(TensorShape(4U, 3U, 2U, 2U), TensorShape(4U, 2U), TensorShape(3U, 2U), TensorShape(4U, 3U, 2U, 2U));
        add_config(TensorShape(17U, 4U, 3U, 2U, 2U), TensorShape(17U, 4U, 4U), TensorShape(3U, 4U),
                   TensorShape(17U, 4U, 3U, 2U, 2U));
        add_config(TensorShape(10U, 4U, 5U, 3U, 2U, 2U), TensorShape(10U, 4U, 5U, 3U), TensorShape(3U, 3U),
                   TensorShape(10U, 4U, 5U, 3U, 2U, 2U));

        // index length is 4
        add_config(TensorShape(35U, 4U, 3U, 2U, 2U), TensorShape(35U, 4U), TensorShape(4U, 4U),
                   TensorShape(35U, 4U, 3U, 2U, 2U));
        add_config(TensorShape(10U, 4U, 5U, 3U, 2U, 2U), TensorShape(10U, 4U, 3U), TensorShape(4U, 3U),
                   TensorShape(10U, 4U, 5U, 3U, 2U, 2U));

        // index length is 5
        add_config(TensorShape(10U, 4U, 5U, 3U, 2U, 2U), TensorShape(10U, 3U), TensorShape(5U, 3U),
                   TensorShape(10U, 4U, 5U, 3U, 2U, 2U));
    }
};

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