diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-11-03 08:15:27 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-03 09:22:57 -0700 |
commit | 1a7c7144669657d90ce5339b1a48cee0f8f95439 (patch) | |
tree | 89f5f657a0d12d9cffb64d2e90eba471d4976ae9 /tensorflow/cc/BUILD | |
parent | f19dc6c5e28694bf067d0bbc9698b57ec4068b4e (diff) |
C++ Gradients: Adds gradient functions for some nn ops. Adds functionality to gradient checker to enable explicit initial values for some tensors.
Change: 138078095
Diffstat (limited to 'tensorflow/cc/BUILD')
-rw-r--r-- | tensorflow/cc/BUILD | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index a804955a36..39d519a707 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -264,6 +264,36 @@ tf_cc_test( ], ) +cc_library( + name = "nn_grad", + srcs = ["gradients/nn_grad.cc"], + deps = [ + ":cc_ops", + ":grad_op_registry", + ":ops", + ":scope", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + ], +) + +tf_cc_test( + name = "gradients_nn_grad_test", + srcs = ["gradients/nn_grad_test.cc"], + deps = [ + ":cc_ops", + ":grad_op_registry", + ":grad_testutil", + ":gradient_checker", + ":nn_grad", + ":testutil", + "//tensorflow/core:lib_internal", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + tf_gen_op_wrappers_cc( name = "cc_ops", op_lib_names = [ |