Function get_matmul_opr

dnn/src/fallback/convolution/algos.cpp:29–47

}

MatrixMul* get_matmul_opr(const NCBKernSizeParam& param) {
    using ConvCM = param::Convolution::ComputeMode;
    using MmCM = param::MatrixMul::ComputeMode;
    static CpuOprDelegationStorage<3> storage;
    if (param.filter_meta.format == param::Convolution::Format::NCHW44) {
        MatrixMul::Param p;
        p.format = param::MatrixMul::Format::MK4;
        return storage.get<MatrixMul, 0>(p);
    }
    switch (param.compute_mode) {
        default:
            return storage.get<MatrixMul, 1>({});
        case ConvCM::FLOAT32: {
            MatrixMul::Param p;
            p.compute_mode = MmCM::FLOAT32;
            return storage.get<MatrixMul, 2>(p);
        }
    }
}

WorkspaceBundle get_bundle(const NCBKernSizeParam& param) {
    UNPACK_CONV_F32_NCB_KERN_SIZES(param);