由于工作需要,需要用cudnn实现tensor add kernel.
代码传不了,只能传图片
头文件:
cpp文件
测试文件:
deep learning张量加法实现之cudnn
最新推荐文章于 2022-04-18 23:25:55 发布
由于工作需要,需要用cudnn实现tensor add kernel.
代码传不了,只能传图片
头文件:
cpp文件
测试文件: