diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index ccb12d9b09d..3653e59ff08 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -1743,6 +1743,7 @@ tf_cuda_cc_test( "//tensorflow/core:test", "//tensorflow/core:test_main", "//tensorflow/core:testlib", + "//tensorflow/core/platform:tf32_utils", "@com_google_absl//absl/algorithm:container", ], )