diff options
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 = [ |