Adding compatible_with attribute to tflite custom kernels

PiperOrigin-RevId: 360341636 Change-Id: I903868ff92e6c41d206acbe17e9838c2c5ad21e3
2021-03-01 20:52:52 -08:00 · 2021-03-01 20:52:52 -08:00 · b9ce1f83a7
commit b9ce1f83a7
parent 803f57cafd
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/lite/kernels/BUILD
+++ b/tensorflow/lite/kernels/BUILD
@ -743,6 +743,7 @@ cc_library(
        "random_uniform.cc",
    ],
    hdrs = ["custom_ops_register.h"],
+    compatible_with = get_compatible_with_portable(),
    copts = tflite_copts(),
    deps = [
        ":kernel_util",