bazel_tf_op:向tensorflow添加自定义操作

时间:2024-05-21 10:54:10
【文件属性】:

文件名称:bazel_tf_op:向tensorflow添加自定义操作

文件大小:18KB

文件格式:ZIP

更新时间:2024-05-21 10:54:10

Python

bazel_tf_op add a custom tensorflow op in bazel 如何建造 bazel build -c opt --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" op:matrix_add_op python op/test_matrix_add.py


【文件预览】:
bazel_tf_op-master
----configure.py(735B)
----README.md(187B)
----tests()
--------tests.cc(511B)
--------BUILD(374B)
----cuda.BUILD(1KB)
----gtest.BUILD(415B)
----tensorflow_config.txt(254B)
----op()
--------configure.py(735B)
--------matrix_add_op.h(975B)
--------__init__.pyc(930B)
--------__init__.py(714B)
--------BUILD(481B)
--------matrix_add_kernel.cc(2KB)
--------matrix_add.cc(2KB)
--------matrix_add_kernel.cu(3KB)
--------test_matrix_add.py(2KB)
--------matrix_add_op.cc(3KB)
----tf.BUILD(407B)
----crosstool()
--------CROSSTOOL(9KB)
--------osx_wrapper.sh(4KB)
--------crosstool_wrapper_driver_is_not_gcc(9KB)
--------BUILD(595B)
--------crosstool_ar(446B)
----WORKSPACE(493B)

网友评论