Function af_var_v2

src/api/c/var.cpp:194–253

}

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

        af_array output       = 0;
        const ArrayInfo& info = getInfo(in);
        af_dtype type         = info.getType();

        af_array no_weights = 0;
        switch (type) {
            case f32:
                output = var_<float, float>(in, no_weights, bias, dim);
                break;
            case f64:
                output = var_<double, double>(in, no_weights, bias, dim);
                break;
            case s32:
                output = var_<int, float>(in, no_weights, bias, dim);
                break;
            case u32:
                output = var_<uint, float>(in, no_weights, bias, dim);
                break;
            case s16:
                output = var_<short, float>(in, no_weights, bias, dim);
                break;
            case u16:
                output = var_<ushort, float>(in, no_weights, bias, dim);
                break;
            case s64:
                output = var_<intl, double>(in, no_weights, bias, dim);
                break;
            case u64:
                output = var_<uintl, double>(in, no_weights, bias, dim);
                break;
            case s8:
                output = var_<schar, float>(in, no_weights, bias, dim);
                break;
            case u8:
                output = var_<uchar, float>(in, no_weights, bias, dim);
                break;
            case b8:
                output = var_<char, float>(in, no_weights, bias, dim);
                break;
            case c32:
                output = var_<cfloat, cfloat>(in, no_weights, bias, dim);
                break;
            case c64:
                output = var_<cdouble, cdouble>(in, no_weights, bias, dim);
                break;
            case f16:
                output = var_<half, half>(in, no_weights, bias, dim);
                break;
            default: TYPE_ERROR(1, type);
        }
        std::swap(*out, output);
    }
    CATCHALL;