Question

我使用conv3d构建了张量流模型，将其冻结并针对推理进行了优化。当我在Android（TensorflowInferenceInterface.run）中调用推理时，我收到以下错误：

java.lang.IllegalArgumentException: No OpKernel was registered to support
Op 'Conv3D' with these attrs.  Registered devices: [CPU], Registered kernels:
   <no registered kernels>
   [[Node: conv3d/convolution = Conv3D[T=DT_FLOAT, data_format="NDHWC", padding="VALID", strides=[1, 1, 1, 1, 1]](images, conv3d/kernel/read)]]

据我所知，包括Conv3D在内的一些操作不会添加到tensorflow推理库中以节省空间。

因此，我使用tensorflow/python/tools/print_selective_registration_header.py脚本生成在我的图表上运行推理所需的操作集，将输出的ops_to_register.h放到core/framework，然后使用

bazel build -c opt --copt="-DSELECTIVE_REGISTRATION" \
--copt="-DSUPPORT_SELECTIVE_REGISTRATION" \
//tensorflow/contrib/android:libtensorflow_inference.so \
--host_crosstool_top=@bazel_tools//tools/cpp:toolchain \
--crosstool_top=//external:android/crosstool --cpu=armeabi-v7a

将libtensorflow_inference.so复制到我的app/src/main/jniLibs文件夹，但错误（Conv3D未注册）仍然存在。你知道出了什么问题吗？

Answer 1

看起来创建ops_to_register.h还不够。我必须修改tensorflow/core/kernels/BUILD文件，以便包含tensorflow/core/kernels/的文件列在android_extended_ops_group1或android_extended_ops_group2部分（以及.h文件中android_extended_ops_headers 1}}部分。

Conv3D无法正常工作

1 个答案: