Function prepare_sub_opr

dnn/src/cambricon/convolution/forward/algo.cpp:40–47

}

std::pair<TensorLayoutArray, std::unique_ptr<ConvBiasForward>> prepare_sub_opr(
        const ConvolutionForwardImpl::AlgoBase::SizeArgs& args) {
    auto conv_bias_opr = args.opr->handle()->create_operator<ConvBiasForward>();
    auto&& config = sub_opr_config(
            *args.layout_src, *args.layout_filter, *args.layout_dst, args.opr);
    conv_bias_opr->param() = config.second;
    return {config.first, std::move(conv_bias_opr)};
}

WorkspaceBundle get_workspace_bundle(
        const std::pair<TensorLayoutArray, std::unique_ptr<ConvBiasForward>>&