| Advait Jain | 1bdbbcb | 2021-05-17 16:46:02 +0000 | [diff] [blame] | 1 | diff --git a/tensorflow/lite/micro/kernels/BUILD b/tensorflow/lite/micro/kernels/BUILD |
| 2 | index 22bcf2b1efd..6f27814f6a2 100644 |
| 3 | --- a/tensorflow/lite/micro/kernels/BUILD |
| 4 | +++ b/tensorflow/lite/micro/kernels/BUILD |
| 5 | @@ -1,7 +1,5 @@ |
| 6 | -load( |
| 7 | - "//tensorflow/lite/micro:build_def.bzl", |
| 8 | - "micro_copts", |
| 9 | -) |
| 10 | +load("//tensorflow/lite/micro:build_def.bzl", "micro_copts") |
| 11 | +load("//tensorflow:extra_rules.bzl", "tflm_kernel_friends") |
| 12 | |
| 13 | package( |
| 14 | features = ["-layering_check"], |
| 15 | @@ -23,6 +21,11 @@ package_group( |
| 16 | packages = ["//tensorflow/lite/micro"], |
| 17 | ) |
| 18 | |
| 19 | +package_group( |
| 20 | + name = "kernel_friends", |
| 21 | + packages = tflm_kernel_friends(), |
| 22 | +) |
| 23 | + |
| 24 | #################################### |
| 25 | # C++ libraries |
| 26 | #################################### |
| 27 | @@ -245,6 +248,7 @@ cc_library( |
| 28 | ], |
| 29 | hdrs = ["kernel_util.h"], |
| 30 | visibility = [ |
| 31 | + ":kernel_friends", |
| 32 | ":micro", |
| 33 | ], |
| 34 | deps = [ |
