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/c/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/c/BUILD')
| -rw-r--r-- | tensorflow/c/BUILD | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 6e39deee63..af96ce70b6 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -38,6 +38,9 @@ tf_cuda_library( ], "//conditions:default": [ "//tensorflow/cc/saved_model:loader", + "//tensorflow/cc:gradients", + "//tensorflow/cc:ops", + "//tensorflow/cc:scope_internal", "//tensorflow/core:core_cpu", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -97,21 +100,22 @@ tf_cc_test( # linkstatic = tf_kernel_tests_linkstatic(), deps = [ ":c_api", + "//tensorflow/cc:cc_ops", + "//tensorflow/cc:grad_ops", "//tensorflow/cc/saved_model:signature_constants", "//tensorflow/cc/saved_model:tag_constants", "//tensorflow/core:core_cpu_internal", "//tensorflow/core:direct_session", "//tensorflow/core:framework", + "//tensorflow/core:framework_internal", "//tensorflow/core:lib", "//tensorflow/core:proto_text", "//tensorflow/core:protos_all_cc", "//tensorflow/core:test", "//tensorflow/core:test_main", - "//tensorflow/core:testlib", "//tensorflow/core/kernels:array", "//tensorflow/core/kernels:control_flow_ops", "//tensorflow/core/kernels:math", - "//third_party/eigen3", ], ) |