iOS的TensorFlow是否支持FIFOQueue?

时间:2017-02-04 10:12:50

标签: tensorflow

适用于iOS的TensorFlow包含FIFOQueue吗?

我训练了一个模型,冻结了它,并针对推理进行了优化,但我在iOS上收到有关FIFOQueue的错误消息

"无法创建TensorFlow图:无效参数:未注册OpKernel以支持Op' FIFOQueue'与这些attrs。已注册的设备:[CPU],已注册的内核: [[Node:fifo_queue = FIFOQueuecapacity = 30,component_types = [DT_FLOAT,DT_FLOAT,DT_FLOAT],container ="",shapes = [[640,480,3],[300,1,2],[300 ,1,4]],shared_name =""]]"

我甚至尝试在训练和获取input_graph.pb之前在tensorflow / contrib / slim / python / slim / nets / inception_v1.py中设置is_training = False。 (我尝试使用is_training = True创建的检查点)

TensorFlow for iOS不支持FIFOQueue吗?

1 个答案:

答案 0 :(得分:0)

我添加了" tensorflow / core / kernels / fifo_queue_op.cc"到tensorflow / contrib / makefile / tf_op_files.txt然后运行compile_ios_tensorflow.sh,错误就消失了。