文件名称：custom-op:TensorFlow的自定义操作指南
文件大小：100KB
文件格式：ZIP
更新时间：2024-05-24 17:19:04
Smarty
TensorFlow自定义操作 本指南适用于想要为TensorFlow编写自定义c ++ op并将op作为pip软件包分发的用户。 该存储库既是操作构建和打包过程的工作示例，又是编写自己的操作的模板/起点。 该存储库的设置方式使您可以从TensorFlow的pip包中构建自定义操作，而不是从头开始构建TensorFlow。 这保证了您构建的共享库将与TensorFlow的pip软件包二进制兼容。 本指南目前支持Ubuntu和Windows自定义操作，并且包括cpu和gpu操作的示例。 从2019年8月1日开始，每晚tf-nightly和tf-nightly-gpu预览以及官方发行版tensorflow和tensorflow-gpu 1.14.0之前的版本现在使用不同的环境构建（Ubuntu 16.04与Ubuntu 14.04相比，适用于例如），这是我们努力使TensorFlow的pi
【文件预览】：
custom-op-master
----MANIFEST.in(86B)
----tensorflow_time_two()
--------BUILD(4KB)
--------python()
--------cc()
--------__init__.py(835B)
----configure.sh(8KB)
----BUILD(246B)
----tensorflow_zero_out()
--------BUILD(2KB)
--------python()
--------cc()
--------__init__.py(831B)
----build_pip_pkg.sh(2KB)
----LICENSE(11KB)
----gpu()
--------cuda()
--------cuda_configure.bzl(44KB)
--------BUILD(0B)
--------crosstool()
--------find_cuda_config.py(17KB)
----setup.py(3KB)
----third_party()
--------toolchains()
----Makefile(2KB)
----tf()
--------BUILD(0B)
--------tf_configure.bzl(7KB)
--------BUILD.tpl(439B)
----README.md(19KB)
----WORKSPACE(180B)