Enable Conv3D op for android flex delegate

PiperOrigin-RevId: 307501841 Change-Id: Icbeb0fd9d2fc056f4e96910f96b2586a6944216e
2020-04-20 16:42:50 -07:00 · 2020-04-20 16:42:50 -07:00 · e7046ce5d6
commit e7046ce5d6
parent 8f439272ed
1 changed files with 5 additions and 0 deletions
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@ -6991,6 +6991,11 @@ filegroup(
        # rule above. Seems to have only have worked before because of
        # hdrs_check loose.
        "stateful_random_ops_cpu_gpu.h",
+        # Allows conv_3d ops for android but excluded from *_3d* rule above.
+        "conv_3d.h",
+        "conv_ops_3d.h",
+        "conv_ops_3d.cc",
+        "conv_ops_gpu.h",
    ],
    visibility = ["//visibility:public"],
 )