Function af_var_weighted

src/api/c/var.cpp:255–335

}

af_err af_var_weighted(af_array* out, const af_array in, const af_array weights,
                       const dim_t dim) {
    try {
        ARG_ASSERT(3, (dim >= 0 && dim <= 3));

        af_array output        = 0;
        const ArrayInfo& iInfo = getInfo(in);
        const ArrayInfo& wInfo = getInfo(weights);
        af_dtype iType         = iInfo.getType();
        af_dtype wType         = wInfo.getType();

        ARG_ASSERT(
            2,
            (wType == f32 ||
             wType ==
                 f64)); /* verify that weights are non-complex real numbers */

        switch (iType) {
            case f64:
                output = var_<double, double>(in, weights,
                                              AF_VARIANCE_POPULATION, dim);
                break;
            case f32:
                output = var_<float, float>(in, weights, AF_VARIANCE_POPULATION,
                                            dim);
                break;
            case s32:
                output =
                    var_<int, float>(in, weights, AF_VARIANCE_POPULATION, dim);
                break;
            case u32:
                output =
                    var_<uint, float>(in, weights, AF_VARIANCE_POPULATION, dim);
                break;
            case s16:
                output = var_<short, float>(in, weights, AF_VARIANCE_POPULATION,
                                            dim);
                break;
            case u16:
                output = var_<ushort, float>(in, weights,
                                             AF_VARIANCE_POPULATION, dim);
                break;
            case s64:
                output = var_<intl, double>(in, weights, AF_VARIANCE_POPULATION,
                                            dim);
                break;
            case u64:
                output = var_<uintl, double>(in, weights,
                                             AF_VARIANCE_POPULATION, dim);
                break;
            case s8:
                output = var_<schar, float>(in, weights, AF_VARIANCE_POPULATION,
                                            dim);
                break;
            case u8:
                output = var_<uchar, float>(in, weights, AF_VARIANCE_POPULATION,
                                            dim);
                break;