diff options
author | Jingyue Wu <jingyue@google.com> | 2017-03-01 12:50:01 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-01 13:04:50 -0800 |
commit | 6d8cb080fe46586b3063c0e50dc659c42c31f7f5 (patch) | |
tree | 040cf559961085fd875e848eee5bda1e47ebacd8 | |
parent | 03f1ff73c7caf58cd46cbc5d3713eea63dca2246 (diff) |
Fix cmake windows build.
Change: 148919678
-rw-r--r-- | tensorflow/contrib/cmake/tf_core_ops.cmake | 1 | ||||
-rw-r--r-- | tensorflow/contrib/cmake/tf_python.cmake | 8 |
2 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_core_ops.cmake b/tensorflow/contrib/cmake/tf_core_ops.cmake index 763b77dc42..56072fbdd7 100644 --- a/tensorflow/contrib/cmake/tf_core_ops.cmake +++ b/tensorflow/contrib/cmake/tf_core_ops.cmake @@ -48,6 +48,7 @@ GENERATE_CONTRIB_OP_LIBRARY(cudnn_rnn "${tensorflow_source_dir}/tensorflow/contr GENERATE_CONTRIB_OP_LIBRARY(factorization_clustering "${tensorflow_source_dir}/tensorflow/contrib/factorization/ops/clustering_ops.cc") GENERATE_CONTRIB_OP_LIBRARY(factorization_factorization "${tensorflow_source_dir}/tensorflow/contrib/factorization/ops/factorization_ops.cc") GENERATE_CONTRIB_OP_LIBRARY(framework_variable "${tensorflow_source_dir}/tensorflow/contrib/framework/ops/variable_ops.cc") +GENERATE_CONTRIB_OP_LIBRARY(memory_stats "${tensorflow_source_dir}/tensorflow/contrib/memory_stats/ops/memory_stats_ops.cc") GENERATE_CONTRIB_OP_LIBRARY(tensor_forest "${tensorflow_source_dir}/tensorflow/contrib/tensor_forest/ops/tensor_forest_ops.cc") ######################################################## diff --git a/tensorflow/contrib/cmake/tf_python.cmake b/tensorflow/contrib/cmake/tf_python.cmake index cf17c6f666..07374e33fa 100644 --- a/tensorflow/contrib/cmake/tf_python.cmake +++ b/tensorflow/contrib/cmake/tf_python.cmake @@ -337,6 +337,12 @@ add_python_module("tensorflow/contrib/losses/python") add_python_module("tensorflow/contrib/losses/python/losses") add_python_module("tensorflow/contrib/makefile") add_python_module("tensorflow/contrib/makefile/test") +add_python_module("tensorflow/contrib/memory_stats") +add_python_module("tensorflow/contrib/memory_stats/kernels") +add_python_module("tensorflow/contrib/memory_stats/ops") +add_python_module("tensorflow/contrib/memory_stats/python") +add_python_module("tensorflow/contrib/memory_stats/python/kernel_tests") +add_python_module("tensorflow/contrib/memory_stats/python/ops") add_python_module("tensorflow/contrib/metrics") add_python_module("tensorflow/contrib/metrics/kernels") add_python_module("tensorflow/contrib/metrics/ops") @@ -528,6 +534,8 @@ GENERATE_PYTHON_OP_LIB("contrib_factorization_factorization_ops" DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/contrib/factorization/python/ops/gen_factorization_ops.py) GENERATE_PYTHON_OP_LIB("contrib_framework_variable_ops" DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/contrib/framework/python/ops/gen_variable_ops.py) +GENERATE_PYTHON_OP_LIB("contrib_memory_stats_ops" + DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/contrib/memory_stats/ops/gen_memory_stats_ops.py) GENERATE_PYTHON_OP_LIB("contrib_tensor_forest_ops" DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/contrib/tensor_forest/python/ops/gen_tensor_forest_ops.py) |