我正在尝试构建一个在TensorFlow 1.6中使用DatasetOpKernel
的TensorFlow扩展。我是在Ubuntu 16.04上构建的。
奇怪的是,TensorFlow 1.6中的共享库中不存在DatasetOpKernel
符号,尽管它们适用于TensorFlow 1.7。由于与DatasetOpKernel相关的符号未找到错误,我的插件无法加载,并且我已经使用nm确认TensorFlow提供的共享库中没有DatasetOpKernel符号。
我正在使用pip install tensorflow==1.6.0
期间安装的共享库
DatasetOpKernel
显然是TensorFlow 1.6(the relevant section on documentation)的一部分,所以我想知道如何在我的扩展程序中正确链接到它。
答案 0 :(得分:0)
TensorFlow 1.6二进制版本不支持创建自定义DatasetOpKernel
库并使用tf.load_op_library()
机制加载它们。这种支持是added in TensorFlow 1.7。
有关如何创建DatasetOpKernel
库的详细信息,请参阅updated documentation for handling new data formats。