如标题中所述,我无法使用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.py
,SqrtGrad
确实存在。由于在Python上运行模型不会返回此错误,因此当我使用tf.train.write_graph(sess.graph_def, 'location', 'trained_model.pb', as_text=False)
生成protobuf文件时,这是否会出现问题?