diff --git a/tensorflow/lite/micro/kernels/BUILD b/tensorflow/lite/micro/kernels/BUILD index 2a0a2d9e58d..a925bc7dd15 100644 --- a/tensorflow/lite/micro/kernels/BUILD +++ b/tensorflow/lite/micro/kernels/BUILD @@ -179,6 +179,9 @@ cc_library( "kernel_runner.cc", ], hdrs = ["kernel_runner.h"], + visibility = [ + "//learning/brain/contrib/micro/tflite:__pkg__", + ], deps = [ "//tensorflow/lite/c:common", "//tensorflow/lite/kernels/internal:compatibility", diff --git a/tensorflow/lite/micro/testing/BUILD b/tensorflow/lite/micro/testing/BUILD index 335953d1412..71e503abafe 100644 --- a/tensorflow/lite/micro/testing/BUILD +++ b/tensorflow/lite/micro/testing/BUILD @@ -23,6 +23,7 @@ cc_library( visibility = [ ":micro", ":microfrontend", + "//learning/brain/contrib/micro/tflite:__pkg__", ], deps = [ "//tensorflow/lite/c:common",