diff options
| author |  Suharsh Sivakumar <suharshs@google.com> | 2017-04-13 11:42:52 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-13 13:05:08 -0700 |
| commit | 908d5b6ede6ae829dff138a873eec397ef434cd6 (patch) | |
| tree | de7898ec319637d2f6d4a78067715bd02808fb02 /tensorflow/cc/BUILD | |
| parent | 59ccf014e89ff625dc3d9779e1fb54a980c4b6ac (diff) | |
Add C++ gradients to c_api.
#6268
This CL does the following:
(1) Adds TF_AddGradients function to C_API which adds gradient nodes for the specified inputs.
(2) Adds internal constructor for Scope, need to create a scope from an existing graph in the c_api.
(3) Adds constructor for AddSymbolicGradients that assumes OnesLike when grad_inputs aren't provided.
(4) Improves error message when gradients aren't provided.
Change: 153092774
Diffstat (limited to 'tensorflow/cc/BUILD')
| -rw-r--r-- | tensorflow/cc/BUILD | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index aaebdded9a..42fa139282 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -122,7 +122,10 @@ cc_library_with_android_deps( cc_library_with_android_deps( name = "scope", - srcs = ["framework/scope.cc"], + srcs = [ + "framework/scope.cc", + "framework/scope_internal.h", + ], hdrs = ["framework/scope.h"], android_deps = ["//tensorflow/core:android_tensorflow_lib"], common_deps = [ @@ -136,6 +139,15 @@ cc_library_with_android_deps( ], ) +cc_library_with_android_deps( + name = "scope_internal", + hdrs = ["framework/scope_internal.h"], + common_deps = [ + ":scope", + ], + deps = [], +) + tf_cc_test( name = "framework_scope_test", srcs = ["framework/scope_test.cc"], |