diff options
author | 2017-09-07 20:23:33 -0700 | |
---|---|---|
committer | 2017-09-07 20:27:26 -0700 | |
commit | d27db78cd0168f10b308f7508c11dfaa3c6707e9 (patch) | |
tree | cec20fd5c5579626ecb9271e39d1ac59b54dac10 /tensorflow/cc/BUILD | |
parent | 2b5011625bcab6c50c51b948e68063393711bd30 (diff) |
Implement c++ gradients for data_flow operators.
Closes #12856
PiperOrigin-RevId: 167949574
Diffstat (limited to 'tensorflow/cc/BUILD')
-rw-r--r-- | tensorflow/cc/BUILD | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index c6d5792f49..028de60880 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -91,6 +91,7 @@ cc_library( name = "grad_ops", deps = [ ":array_grad", + ":data_flow_grad", ":math_grad", ":nn_grad", ], @@ -363,6 +364,36 @@ tf_cc_test( ], ) +cc_library( + name = "data_flow_grad", + srcs = ["gradients/data_flow_grad.cc"], + deps = [ + ":cc_ops", + ":cc_ops_internal", + ":grad_op_registry", + ":gradients", + ], + alwayslink = 1, +) + +tf_cc_test( + name = "gradients_data_flow_grad_test", + size = "small", + srcs = ["gradients/data_flow_grad_test.cc"], + deps = [ + ":cc_ops", + ":data_flow_grad", + ":grad_op_registry", + ":grad_testutil", + ":gradient_checker", + ":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 = [ |