Function reduce

src/backend/cuda/kernel/reduce.hpp:507–515

template<typename Ti, typename To, af_op_t op>
void reduce(Param<To> out, CParam<Ti> in, int dim, bool change_nan,
            double nanval) {
    switch (dim) {
        case 0: return reduce_first<Ti, To, op>(out, in, change_nan, nanval);
        case 1: return reduce_dim<Ti, To, op, 1>(out, in, change_nan, nanval);
        case 2: return reduce_dim<Ti, To, op, 2>(out, in, change_nan, nanval);
        case 3: return reduce_dim<Ti, To, op, 3>(out, in, change_nan, nanval);
    }
}
template<typename Ti, typename To, af_op_t op>
void reduce_all(Param<To> out, CParam<Ti> in, bool change_nan, double nanval) {
    int in_elements = in.dims[0] * in.dims[1] * in.dims[2] * in.dims[3];