Function prepare_conv_opr

dnn/src/cuda/matrix_mul/conv1x1.cpp:14–36

namespace {

std::unique_ptr<ConvBiasForward> prepare_conv_opr(
        const MatrixMulForwardImpl::AlgoBase::SizeArgs& args) {
    auto conv_bias_opr_ptr = args.opr->handle()->create_operator<ConvBiasForward>();

    auto conv_param_computemode =
            (args.opr->param().compute_mode == param::MatrixMul::ComputeMode::DEFAULT)
                    ? param::Convolution::ComputeMode::DEFAULT
                    : param::Convolution::ComputeMode::FLOAT32;
    conv_bias_opr_ptr->param() = {
            param::ConvBias::NonlineMode::IDENTITY,
            param::Convolution::Mode::CROSS_CORRELATION,
            param::Convolution::Sparse::DENSE,
            param::Convolution::Format::NCHW,
            0,  // pad_h
            0,  // pad_w
            1,  // stride_h
            1,  // stride_w
            1,  // dilate_h
            1,  // dilate_w
            conv_param_computemode};

    return conv_bias_opr_ptr;
}
std::tuple<size_t, size_t, size_t> gen_matrixmul_shape(
        const MatrixMulForwardImpl::AlgoBase::SizeArgs& args) {
    size_t m, k, n;