tensorflow tfcompile-如何生成graph_config.pb?

时间:2018-07-02 12:56:01

标签: tensorflow tensorflow-xla

全部!这些天我遇到了问题。 问题是,使用Tensorflow加速基于ARM的目标的推理。 我正在练习 tfcompile this dude's page引用,我使用:

tfcompile --graph=test_graph.pb --config=test_graph.pbtxt --entry_point=test_func --cpp_class=test --out_object=test_func.o --out_header=test.hpp

test_graph.pbtxt是由以下人员生成的:

user@user $ python
>>>
>>> import tensorflow as tf
>>> from tensorflow.python.platform import gfile
>>> from google.protobuf import text_format
>>> filename='/home/user/test_graph.pb'
>>> with gfile.FastGFile(filename,'rb') as f:
...     graph_def = tf.GraphDef()
...     graph_def.ParseFromString(f.read())
...     tf.import_graph_def(graph_def, name='')
...     tf.train.write_graph(graph_def, '/home/user/', 'test_graph.pbtxt', as_text=True)
>>>

当我运行tfcompile时 我得到了错误:

Non-OK-status: status status: Data loss: Can't parse /home/user/test_graph.pbtxt as text proto

因此* .pbtxt不应从* .pb转换。 我搜索--config=应该是什么,但没有结果。 你能帮我解决吗? 提前谢谢!

0 个答案:

没有答案