Function af_meanvar

src/api/c/var.cpp:423–493

}

af_err af_meanvar(af_array* mean, af_array* var, const af_array in,
                  const af_array weights, const af_var_bias bias,
                  const dim_t dim) {
    try {
        const ArrayInfo& iInfo = getInfo(in);
        if (weights != 0) {
            const ArrayInfo& wInfo = getInfo(weights);
            af_dtype wType         = wInfo.getType();
            ARG_ASSERT(3, (wType == f32 || wType == f64));
        }
        af_dtype iType = iInfo.getType();

        switch (iType) {
            case f32:
                tie(*mean, *var) =
                    meanvar<float, float>(in, weights, bias, dim);
                break;
            case f64:
                tie(*mean, *var) =
                    meanvar<double, double>(in, weights, bias, dim);
                break;
            case s32:
                tie(*mean, *var) = meanvar<int, float>(in, weights, bias, dim);
                break;
            case u32:
                tie(*mean, *var) = meanvar<uint, float>(in, weights, bias, dim);
                break;
            case s16:
                tie(*mean, *var) =
                    meanvar<short, float>(in, weights, bias, dim);
                break;
            case u16:
                tie(*mean, *var) =
                    meanvar<ushort, float>(in, weights, bias, dim);
                break;
            case s64:
                tie(*mean, *var) =
                    meanvar<intl, double>(in, weights, bias, dim);
                break;
            case u64:
                tie(*mean, *var) =
                    meanvar<uintl, double>(in, weights, bias, dim);
                break;
            case s8:
                tie(*mean, *var) =
                    meanvar<schar, float>(in, weights, bias, dim);
                break;
            case u8:
                tie(*mean, *var) =
                    meanvar<uchar, float>(in, weights, bias, dim);
                break;
            case b8:
                tie(*mean, *var) = meanvar<char, float>(in, weights, bias, dim);
                break;
            case c32:
                tie(*mean, *var) =
                    meanvar<cfloat, cfloat>(in, weights, bias, dim);
                break;