Method get_workspace_in_bytes

dnn/src/fallback/mask_conv/opr_impl.cpp:188–197

}

size_t MaskConvForwardImpl::get_workspace_in_bytes(
        const TensorLayout& src, const TensorLayout& filter, const TensorLayout& mask,
        const TensorLayout& dst) {
    megdnn_ignore(mask);
    size_t OC = filter[0], IC = filter[1], IH = src[2], IW = src[3], OH = dst[2],
           OW = dst[3], FH = filter[2], FW = filter[3], PH = param().pad_h,
           PW = param().pad_w;
    auto wbundle = get_wbundle(OC, OH, OW, IC, IH, IW, FH, FW, PH, PW);
    return wbundle.total_size_in_bytes();
}

}  // namespace fallback
}  // namespace megdnn