Function formatInput

tools/cv/source/imgproc/filter.cpp:23–39

}

static halide_type_t formatInput(VARP& src, bool fp = true) {
    auto info = src->getInfo();
    auto dim = info->dim;
    int height, width, channel;
    getVARPSize(src, &height, &width, &channel);
    if (dim.size() != 4) {
        if (src->getInfo()->order == NHWC) {
            src = _Reshape(src, {1, height, width, channel});
        } else {
            src = _Convert(_Reshape(src, {1, channel, height, width}), NHWC);
        }
    }
    if (fp && src->getInfo() && src->getInfo()->type.code != halide_type_float) {
        src = _Cast(src, halide_type_of<float>());
    }
    return info->type;
}

static VARP formatOutput(VARP src, halide_type_t type) {
    auto dim = src->getInfo()->dim;