tfopgen:为Tensorflow自定义运算符生成C ++和CUDA样板

上传者: 42108948 | 上传时间: 2022-07-19 22:42:45 | 文件大小: 19KB | 文件类型: ZIP
Tensorflow自定义操作员代码大纲生成器 编写张量流运算符需要编写大量的样板C ++和CUDA代码。 该脚本为Tensorflow运算符的CPU和GPU版本生成代码。 更具体地,给定的tensorflow inputs , outputs和attribute s时,产生: C ++头文件,该文件定义操作符类,以Device为模板。 定义操作员的CPU实现的C ++头文件。 具有Shape函数,REGISTER_OP和REGISTER_KERNEL_BUILDER构造的C ++源文件。 Cuda标头,用于定义操作员的GPU实现,包括CUDA内核。 Cuda源文件,带有GPU REGISTER_KERNEL_BUILDER供操作员使用。 python单元测试用例,它构造随机输入数据,并调用运算符。 Makefile,用于使用g ++和nvcc将运算符编译为共享库。 要求

文件下载

资源详情

[{"title":"( 24 个子文件 19KB ) tfopgen:为Tensorflow自定义运算符生成C ++和CUDA样板","children":[{"title":"tfopgen-master","children":[{"title":"setup.py <span style='color:#111;'> 1.06KB </span>","children":null,"spread":false},{"title":".gitignore <span style='color:#111;'> 1.14KB </span>","children":null,"spread":false},{"title":"tfopgen","children":[{"title":"templates","children":[{"title":"Makefile.j2 <span style='color:#111;'> 1.62KB </span>","children":null,"spread":false},{"title":"cuda_source.j2 <span style='color:#111;'> 606B </span>","children":null,"spread":false},{"title":"cpp_header.j2 <span style='color:#111;'> 1.98KB </span>","children":null,"spread":false},{"title":"test_source.j2 <span style='color:#111;'> 2.93KB </span>","children":null,"spread":false},{"title":"cuda_header.j2 <span style='color:#111;'> 4.72KB </span>","children":null,"spread":false},{"title":"cpp_source.j2 <span style='color:#111;'> 2.79KB </span>","children":null,"spread":false},{"title":"main_header.j2 <span style='color:#111;'> 1.09KB </span>","children":null,"spread":false}],"spread":true},{"title":"bin","children":[{"title":"tfopgen <span style='color:#111;'> 76B </span>","children":null,"spread":false}],"spread":true},{"title":"__init__.py <span style='color:#111;'> 56B </span>","children":null,"spread":false},{"title":"create_op.py <span style='color:#111;'> 4.88KB </span>","children":null,"spread":false},{"title":"examples","children":[{"title":"complex_phase.yml <span style='color:#111;'> 608B </span>","children":null,"spread":false},{"title":"simple.yml <span style='color:#111;'> 227B </span>","children":null,"spread":false}],"spread":true},{"title":"util.py <span style='color:#111;'> 4.15KB </span>","children":null,"spread":false},{"title":"version.py <span style='color:#111;'> 89B </span>","children":null,"spread":false}],"spread":true},{"title":".travis.yml <span style='color:#111;'> 987B </span>","children":null,"spread":false},{"title":"MANIFEST.in <span style='color:#111;'> 101B </span>","children":null,"spread":false},{"title":"install","children":[{"title":"__init__.py <span style='color:#111;'> 0B </span>","children":null,"spread":false},{"title":"versioning.py <span style='color:#111;'> 1.14KB </span>","children":null,"spread":false}],"spread":true},{"title":"setup.cfg <span style='color:#111;'> 66B </span>","children":null,"spread":false},{"title":"LICENSE.txt <span style='color:#111;'> 1.04KB </span>","children":null,"spread":false},{"title":"README.rst <span style='color:#111;'> 4.92KB </span>","children":null,"spread":false},{"title":"CHANGELOG.rst <span style='color:#111;'> 113B </span>","children":null,"spread":false}],"spread":true}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明