diff options
author | 2017-02-02 16:25:59 -0800 | |
---|---|---|
committer | 2017-02-02 16:44:23 -0800 | |
commit | fe5a99a7ff63f0f68beb9ee89e80dd3fc42a20e1 (patch) | |
tree | c3a2094f03478280b2fac09284b177f1bc92ee26 /tensorflow/contrib/cmake/tf_cc_ops.cmake | |
parent | f0fa1e44376f309886f58934d31fccd0cac7c455 (diff) |
Fix cmake build to reflect recent changes to the C++ API generation script.
Change: 146419788
Diffstat (limited to 'tensorflow/contrib/cmake/tf_cc_ops.cmake')
-rw-r--r-- | tensorflow/contrib/cmake/tf_cc_ops.cmake | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/cmake/tf_cc_ops.cmake b/tensorflow/contrib/cmake/tf_cc_ops.cmake index 1eee73b7e4..6eaa2502be 100644 --- a/tensorflow/contrib/cmake/tf_cc_ops.cmake +++ b/tensorflow/contrib/cmake/tf_cc_ops.cmake @@ -66,12 +66,16 @@ foreach(tf_cc_op_lib_name ${tf_cc_op_lib_names}) add_custom_command( OUTPUT ${cc_ops_target_dir}/${tf_cc_op_lib_name}.h ${cc_ops_target_dir}/${tf_cc_op_lib_name}.cc - COMMAND ${tf_cc_op_lib_name}_gen_cc ${cc_ops_target_dir}/${tf_cc_op_lib_name}.h ${cc_ops_target_dir}/${tf_cc_op_lib_name}.cc ${cc_ops_include_internal} + ${cc_ops_target_dir}/${tf_cc_op_lib_name}_internal.h + ${cc_ops_target_dir}/${tf_cc_op_lib_name}_internal.cc + COMMAND ${tf_cc_op_lib_name}_gen_cc ${cc_ops_target_dir}/${tf_cc_op_lib_name}.h ${cc_ops_target_dir}/${tf_cc_op_lib_name}.cc ${tensorflow_source_dir}/tensorflow/cc/ops/op_gen_overrides.pbtxt ${cc_ops_include_internal} DEPENDS ${tf_cc_op_lib_name}_gen_cc create_cc_ops_header_dir ) list(APPEND tf_cc_ops_generated_files ${cc_ops_target_dir}/${tf_cc_op_lib_name}.h) list(APPEND tf_cc_ops_generated_files ${cc_ops_target_dir}/${tf_cc_op_lib_name}.cc) + list(APPEND tf_cc_ops_generated_files ${cc_ops_target_dir}/${tf_cc_op_lib_name}_internal.h) + list(APPEND tf_cc_ops_generated_files ${cc_ops_target_dir}/${tf_cc_op_lib_name}_internal.cc) endforeach() |