diff --git a/tensorflow/c/eager/BUILD b/tensorflow/c/eager/BUILD index 525fa6d64ba..23cd1c244dc 100644 --- a/tensorflow/c/eager/BUILD +++ b/tensorflow/c/eager/BUILD @@ -459,6 +459,37 @@ tf_cuda_cc_test( ], ) +tf_cuda_cc_test( + name = "mnist_gradients_test", + size = "small", + srcs = [ + "mnist_gradients_test.cc", + ], + args = ["--heap_check=local"], + extra_copts = tfe_xla_copts(), + linkstatic = tf_kernel_tests_linkstatic(), + tags = tf_cuda_tests_tags() + ["nomac"], + deps = [ + ":abstract_tensor_handle", + ":c_api_experimental", + ":c_api_test_util", + ":c_api_unified_internal", + ":gradients_internal", + "//tensorflow/c:c_api", + "//tensorflow/c:c_test_util", + "//tensorflow/c:tf_status_helper", + "//tensorflow/cc/profiler", + "//tensorflow/compiler/mlir/tensorflow/c:mlir_c_api_registration", + "//tensorflow/core:lib", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core/lib/llvm_rtti", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", + ], +) + cc_library( name = "abstract_tensor_handle", hdrs = ["abstract_tensor_handle.h"],