未找到:在Android上创建Tensorflow图时,操作类型未注册'SqrtGrad'

时间:2016-11-07 12:13:11

标签: android java-native-interface tensorflow cc

如标题中所述,我无法使用JNI从Android上的导入模型创建Tensorflow图。尝试运行tensorflow::Status s = session->Create(graph_def);时,状态将返回Not found: Op type not registered 'SqrtGrad'

我检查了jni-build/jni/include/tensorflow/python/ops/math_ops.pySqrtGrad确实存在。由于在Python上运行模型不会返回此错误,因此当我使用tf.train.write_graph(sess.graph_def, 'location', 'trained_model.pb', as_text=False)生成protobuf文件时,这是否会出现问题?

0 个答案:

没有答案