diff options
author | 2017-07-13 07:40:00 -0700 | |
---|---|---|
committer | 2017-07-13 07:44:31 -0700 | |
commit | 31f1375c01782f53cf12ce2aa4efa584483c14ea (patch) | |
tree | a5422d5a1823fb892fb0aceb4272464b134d2ed9 /tensorflow/contrib/cmake/tf_core_ops.cmake | |
parent | 1eccfb6e190cdd594c2cd1d3b27323f87614d8bc (diff) |
Add GradientsDebugger to tfdbg
to allow retrieval of gradient tensors created by TensorFlow's automatic differentiation algorithm (i.e., tf.gradients and optimizer code that uses it).
PiperOrigin-RevId: 161805516
Diffstat (limited to 'tensorflow/contrib/cmake/tf_core_ops.cmake')
-rw-r--r-- | tensorflow/contrib/cmake/tf_core_ops.cmake | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_core_ops.cmake b/tensorflow/contrib/cmake/tf_core_ops.cmake index 21d94da6cc..b350b822dd 100644 --- a/tensorflow/contrib/cmake/tf_core_ops.cmake +++ b/tensorflow/contrib/cmake/tf_core_ops.cmake @@ -129,3 +129,15 @@ list(REMOVE_ITEM tf_core_ops_srcs ${tf_core_ops_exclude_srcs}) add_library(tf_core_ops OBJECT ${tf_core_ops_srcs}) add_dependencies(tf_core_ops tf_core_cpu) + +######################################################## +# tf_debug_ops library +######################################################## + +file(GLOB tf_debug_ops_srcs + "${tensorflow_source_dir}/tensorflow/core/ops/debug_ops.cc" +) + +add_library(tf_debug_ops OBJECT ${tf_debug_ops_srcs}) + +add_dependencies(tf_debug_ops tf_core_framework) |