experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable gradient_checker_test on TAP.

Browse Source 
PiperOrigin-RevId: 347087545
Change-Id: I643fe65ff0eeb59c193e5f1d8faa337ce4f26c9a
This commit is contained in:
Saurabh Saxena 2020-12-11 15:30:07 -08:00 committed by TensorFlower Gardener
parent 6d8366afc4
commit 5a51f9bbed
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/c/eager/BUILD
View File

@ -400,8 +400,9 @@ cc_library(
"//tensorflow:internal",
],
deps = [
":abstract_tensor_handle",
":unified_api_testutil",
"//tensorflow/c/eager:abstract_tensor_handle",
"//tensorflow/c:tf_tensor_internal",
"//tensorflow/c/experimental/ops:math_ops",
"@com_google_absl//absl/types:span",
],
@ -417,14 +418,15 @@ tf_cuda_cc_test(
linkstatic = tf_kernel_tests_linkstatic(),
tags = tf_cuda_tests_tags() + [
"no_cuda_asan", # b/175330074
"notap", # b/175330074
],
deps = [
":abstract_tensor_handle",
":c_api_experimental",
":gradient_checker",
":unified_api_testutil",
"//tensorflow/c:tf_status_helper",
"//tensorflow/c/eager:abstract_tensor_handle",
"//tensorflow/c/eager:c_api_experimental",
"//tensorflow/c:tf_tensor_internal",
"//tensorflow/c/experimental/ops",
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"@com_google_absl//absl/types:span",