如何向tensorflow添加自定义数据类型?

时间:2018-06-25 03:55:07

标签: c++ tensorflow

我有重载算术运算的float24数据类型(以32位存储)的C ++实现。我想将此类型添加到tensorflow C ++和Python API中,以便在仅在CPU上运行的tf图中使用它。

我没有找到如何操作的指南。我应该在哪里注册我的类型?还会生成python代码吗?

0 个答案:

没有答案