我正在开发uTensor
及其代码生成器utensor_cgen
。
https://github.com/uTensor/uTensor
https://github.com/uTensor/utensor_cgen
对于代码生成器,我必须解析量化的graph_def
协议缓冲文件(pb file
)。
但是,我发现有些输入节点在tensorboard
中不可见,但在graph_def.node
中可见。
他们的名字以^
开头,当我查看tf.import_graph_def
源代码时,我发现了这个_IsControlInput
函数:
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/framework/importer.py#L92
我真的需要知道“控制输入”的定义,这样我才能正确解析pb文件并将其转换为针对MCU的正确uTensor
实现。
有人可以告诉我在哪里可以找到这个定义吗? 似乎没有在文档中找到。
感谢。
答案 0 :(得分:0)