diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 4ad69ae3fbd..3ab4e8efcdb 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -58,6 +58,7 @@ tf_cuda_library( "//tensorflow/cc/saved_model:loader", "//tensorflow/cc:gradients", "//tensorflow/cc:ops", + "//tensorflow/cc:grad_ops", "//tensorflow/cc:scope_internal", "//tensorflow/core:core_cpu", "//tensorflow/core:framework", diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index 8810b8731ae..8d4260a0b9c 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -91,6 +91,7 @@ cc_library( deps = [ ":array_grad", ":math_grad", + ":nn_grad", ], )