Function conv_image

test/arrayfire_test.cpp:1357–1390

// TODO: perform conversion on device for CUDA and OpenCL
template<typename T>
af_err conv_image(af_array *out, af_array in) {
    af_array outArray;

    dim_t d0, d1, d2, d3;
    af_get_dims(&d0, &d1, &d2, &d3, in);
    af::dim4 idims(d0, d1, d2, d3);

    dim_t nElems = 0;
    af_get_elements(&nElems, in);

    float *in_data = new float[nElems];
    af_get_data_ptr(in_data, in);

    T *out_data = new T[nElems];

    af_dtype out_type = (af_dtype)af::dtype_traits<T>::af_type;
    for (int i = 0; i < (int)nElems; i++) {
        if (out_type == s8) {
            // shift to avoid overflow
            out_data[i] = (T)(std::trunc(in_data[i]) - 128.f);
        } else {
            out_data[i] = (T)in_data[i];
        }
    }

    af_create_array(&outArray, out_data, idims.ndims(), idims.get(), out_type);

    std::swap(*out, outArray);

    delete[] in_data;
    delete[] out_data;

    return AF_SUCCESS;
}

#define INSTANTIATE(To) \
    template af_err conv_image<To>(af_array * out, af_array in)