Include c++ gradients in c_api build rule.
#6268 #9150 Change: 155146664
This commit is contained in:
parent
ffc5e6fbf6
commit
81fbb1246c
@ -58,6 +58,7 @@ tf_cuda_library(
|
|||||||
"//tensorflow/cc/saved_model:loader",
|
"//tensorflow/cc/saved_model:loader",
|
||||||
"//tensorflow/cc:gradients",
|
"//tensorflow/cc:gradients",
|
||||||
"//tensorflow/cc:ops",
|
"//tensorflow/cc:ops",
|
||||||
|
"//tensorflow/cc:grad_ops",
|
||||||
"//tensorflow/cc:scope_internal",
|
"//tensorflow/cc:scope_internal",
|
||||||
"//tensorflow/core:core_cpu",
|
"//tensorflow/core:core_cpu",
|
||||||
"//tensorflow/core:framework",
|
"//tensorflow/core:framework",
|
||||||
|
@ -91,6 +91,7 @@ cc_library(
|
|||||||
deps = [
|
deps = [
|
||||||
":array_grad",
|
":array_grad",
|
||||||
":math_grad",
|
":math_grad",
|
||||||
|
":nn_grad",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user