diff --git a/tensorflow/lite/delegates/gpu/metal/kernels/BUILD b/tensorflow/lite/delegates/gpu/metal/kernels/BUILD index 17e59e70ebe..421ddbb7f83 100644 --- a/tensorflow/lite/delegates/gpu/metal/kernels/BUILD +++ b/tensorflow/lite/delegates/gpu/metal/kernels/BUILD @@ -1,5 +1,5 @@ load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test") -load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite") +load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite", "tflite_ios_per_kernel_test") package( default_visibility = ["//visibility:public"], @@ -509,3 +509,5 @@ objc_library( ) tflite_portable_test_suite() + +tflite_ios_per_kernel_test() diff --git a/tensorflow/lite/special_rules.bzl b/tensorflow/lite/special_rules.bzl index aa3fbfe4985..1c2333c6a75 100644 --- a/tensorflow/lite/special_rules.bzl +++ b/tensorflow/lite/special_rules.bzl @@ -5,6 +5,11 @@ def tflite_portable_test_suite(**kwargs): _ignore = [kwargs] pass +def tflite_ios_per_kernel_test(**kwargs): + """This is a no-op outside of Google.""" + _ignore = [kwargs] + pass + def ios_visibility_whitelist(): """This is a no-op outside of Google.""" pass