diff options
Diffstat (limited to 'tensorflow/cc/BUILD')
-rw-r--r-- | tensorflow/cc/BUILD | 31 |
1 files changed, 30 insertions, 1 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index b0c8cc3d0a..3682ebd943 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -19,13 +19,20 @@ load( cc_library( name = "gradients", - srcs = ["framework/gradients.cc"], + srcs = [ + "framework/gradients.cc", + "framework/while_gradients.cc", + "framework/while_gradients.h", + ], hdrs = ["framework/gradients.h"], deps = [ ":cc_ops", + ":cc_ops_internal", ":grad_op_registry", ":ops", ":scope", + ":scope_internal", + ":while_loop", "//tensorflow/core:core_cpu", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -52,6 +59,28 @@ tf_cc_test( ], ) +tf_cc_test( + name = "framework_while_gradients_test", + size = "small", + srcs = ["framework/while_gradients_test.cc"], + deps = [ + ":cc_ops", + ":client_session", + ":grad_op_registry", + ":grad_ops", + ":gradients", + ":testutil", + ":while_loop", + "//tensorflow/core:all_kernels", + "//tensorflow/core:framework", + "//tensorflow/core:framework_internal", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + cc_library( name = "gradient_checker", srcs = ["framework/gradient_checker.cc"], |