有自定义操作时,tf-mlir-translate无法从graphdef转换为mlir

时间:2020-01-24 02:42:29

标签: tensorflow

具体错误消息是

E tensorflow /编译器/ mlir / tensorflow /翻译/tf_mlir_translate.cc:104] 图形导入失败:找不到:运算符类型未在二进制中注册“ XYZ” 在CWR9BZ2-DT上运行。确保Op和Kernel已注册在 在此过程中运行的二进制文件。请注意,如果您正在加载 保存的图形使用了tf.contrib中的操作,正在访问(例如) tf.contrib.resampler应该在导入图形之前完成,因为 首次访问该模块时会延迟注册contrib ops。

自定义操作XYZ在单独的共享库(.so)中定义。有没有办法告诉tf-mlir-translate使用指定的共享库?

0 个答案:

没有答案