卷积实现之cudnn

    技术2022-07-12  84

    由于工作原因,我需要用cudnn实现卷积。代码在内网,只好上传代码图片。 工程主文件: 工程辅助文件: 辅助文件用于造test case,不是必须文件。 cmake 文件: 卷积核头文件: 卷积核cpp文件: 用卷积核构造卷积:

    Processed: 0.014, SQL: 9