diff options
author | 2018-08-03 17:31:50 -0700 | |
---|---|---|
committer | 2018-08-03 17:31:50 -0700 | |
commit | c4b12fbd0341676375f97f2be8a94d3bfadfb70e (patch) | |
tree | 19b10c46047d1942ca48e6cf6da780edb3dcd7c9 /tensorflow/cc/BUILD | |
parent | f10b2802611be27fcfcefbcfeb508402b44aad8d (diff) | |
parent | a2c479b6fd1ec75f37b8104f53a9f4a286cb72ce (diff) |
Merge pull request #21021 from kbsriram:tf-image-grad
PiperOrigin-RevId: 207350813
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 a98f0b00b2..588a45ea43 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -121,6 +121,7 @@ cc_library( deps = [ ":array_grad", ":data_flow_grad", + ":image_grad", ":math_grad", ":nn_grad", ], @@ -332,6 +333,36 @@ tf_cc_test( ) cc_library( + name = "image_grad", + srcs = ["gradients/image_grad.cc"], + deps = [ + ":cc_ops", + ":cc_ops_internal", + ":grad_op_registry", + ":gradients", + ], + alwayslink = 1, +) + +tf_cc_test( + name = "gradients_image_grad_test", + srcs = ["gradients/image_grad_test.cc"], + deps = [ + ":cc_ops", + ":client_session", + ":grad_op_registry", + ":grad_testutil", + ":gradient_checker", + ":image_grad", + ":testutil", + "//tensorflow/core:lib_internal", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + +cc_library( name = "math_grad", srcs = ["gradients/math_grad.cc"], deps = [ |