Function range

src/backend/oneapi/range.cpp:24–40

namespace oneapi {
template<typename T>
Array<T> range(const dim4& dim, const int seq_dim) {
    // Set dimension along which the sequence should be
    // Other dimensions are simply tiled
    int _seq_dim = seq_dim;
    if (seq_dim < 0) {
        _seq_dim = 0;  // column wise sequence
    }

    if (_seq_dim < 0 || _seq_dim > 3) {
        AF_ERROR("Invalid rep selection", AF_ERR_ARG);
    }

    Array<T> out = createEmptyArray<T>(dim);
    kernel::range<T>(out, _seq_dim);

    return out;
}

#define INSTANTIATE(T) \
    template Array<T> range<T>(const af::dim4& dims, const int seq_dims);