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