Method is_available

dnn/src/cambricon/convolution/forward/algo.cpp:102–113

}

bool ConvolutionForwardImpl::AlgoDefault::is_available(const SizeArgs& args) const {
    if (args.opr->param().format != param::Convolution::Format::NHWC) {
        return false;
    }
    auto config = prepare_sub_opr(args);
    bool valid = config.second.get()
                         ->get_algorithm_info_heuristic(
                                 *args.layout_src, *args.layout_filter, config.first[0],
                                 config.first[1], *args.layout_dst)
                         .valid();
    return valid;
}

size_t ConvolutionForwardImpl::AlgoDefault::get_workspace_in_bytes(
        const SizeArgs& args) const {