Conv3D不工作

Question

我用conv3d建立了一个tensorflow模型，并对其进行了冻结和推理优化。当我在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文件，以便在android_extended_ops_group1或android_extended_ops_group2部分中列出tensorflow/core/kernels/包含的文件(以及android_extended_ops_headers部分中的.h文件)。