diff options
Diffstat (limited to 'tensorflow/contrib/cmake/tf_python.cmake')
| -rw-r--r-- | tensorflow/contrib/cmake/tf_python.cmake | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/tensorflow/contrib/cmake/tf_python.cmake b/tensorflow/contrib/cmake/tf_python.cmake index ce305a4b1e..8672e17443 100644 --- a/tensorflow/contrib/cmake/tf_python.cmake +++ b/tensorflow/contrib/cmake/tf_python.cmake @@ -188,6 +188,7 @@ add_python_module("tensorflow/python/lib") add_python_module("tensorflow/python/lib/core") add_python_module("tensorflow/python/lib/io") add_python_module("tensorflow/python/ops") +add_python_module("tensorflow/python/ops/losses") add_python_module("tensorflow/python/platform") add_python_module("tensorflow/python/platform/default") add_python_module("tensorflow/python/platform/summary") @@ -220,6 +221,7 @@ add_python_module("tensorflow/contrib/bayesflow/examples/reinforce_simple") add_python_module("tensorflow/contrib/bayesflow/python") add_python_module("tensorflow/contrib/bayesflow/python/kernel_tests") add_python_module("tensorflow/contrib/bayesflow/python/ops") +add_python_module("tensorflow/contrib/compiler") add_python_module("tensorflow/contrib/copy_graph") add_python_module("tensorflow/contrib/copy_graph/python") add_python_module("tensorflow/contrib/copy_graph/python/util") @@ -261,6 +263,12 @@ add_python_module("tensorflow/contrib/grid_rnn") add_python_module("tensorflow/contrib/grid_rnn/python") add_python_module("tensorflow/contrib/grid_rnn/python/kernel_tests") add_python_module("tensorflow/contrib/grid_rnn/python/ops") +add_python_module("tensorflow/contrib/image") +add_python_module("tensorflow/contrib/image/python") +add_python_module("tensorflow/contrib/image/python/ops") +add_python_module("tensorflow/contrib/input_pipeline") +add_python_module("tensorflow/contrib/input_pipeline/python") +add_python_module("tensorflow/contrib/input_pipeline/python/ops") add_python_module("tensorflow/contrib/integrate") add_python_module("tensorflow/contrib/integrate/python") add_python_module("tensorflow/contrib/integrate/python/ops") @@ -301,6 +309,7 @@ add_python_module("tensorflow/contrib/learn/python/learn/preprocessing/tests") add_python_module("tensorflow/contrib/learn/python/learn/tests") add_python_module("tensorflow/contrib/learn/python/learn/tests/dataframe") add_python_module("tensorflow/contrib/learn/python/learn/utils") +add_python_module("tensorflow/contrib/legacy_seq2seq") add_python_module("tensorflow/contrib/linalg") add_python_module("tensorflow/contrib/linalg/python") add_python_module("tensorflow/contrib/linalg/python/ops") @@ -392,6 +401,7 @@ add_python_module("tensorflow/contrib/training/python") add_python_module("tensorflow/contrib/training/python/training") add_python_module("tensorflow/contrib/util") + # Additional directories with no Python sources. add_custom_command(TARGET tf_python_touchup_modules PRE_BUILD COMMAND ${CMAKE_COMMAND} -E make_directory "${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/tensorboard/dist") @@ -423,6 +433,7 @@ set(tf_python_op_lib_names ) function(GENERATE_PYTHON_OP_LIB tf_python_op_lib_name) + set(options SHAPE_FUNCTIONS_NOT_REQUIRED) set(oneValueArgs DESTINATION) set(multiValueArgs ADDITIONAL_LIBRARIES) cmake_parse_arguments(GENERATE_PYTHON_OP_LIB @@ -432,7 +443,12 @@ function(GENERATE_PYTHON_OP_LIB tf_python_op_lib_name) set(GENERATE_PYTHON_OP_LIB_DESTINATION "${python_ops_target_dir}/gen_${tf_python_op_lib_name}.py") endif() - + if(GENERATE_PYTHON_OP_LIB_SHAPE_FUNCTIONS_NOT_REQUIRED) + set(require_shape_fn 0) + else() + set(require_shape_fn 1) + endif() + # Create a C++ executable that links in the appropriate op # registrations and generates Python wrapper code based on the # registered ops. @@ -453,7 +469,7 @@ function(GENERATE_PYTHON_OP_LIB tf_python_op_lib_name) # containing the wrappers. add_custom_command( OUTPUT ${GENERATE_PYTHON_OP_LIB_DESTINATION} - COMMAND ${tf_python_op_lib_name}_gen_python @${tensorflow_source_dir}/tensorflow/python/ops/hidden_ops.txt 1 > ${GENERATE_PYTHON_OP_LIB_DESTINATION} + COMMAND ${tf_python_op_lib_name}_gen_python @${tensorflow_source_dir}/tensorflow/python/ops/hidden_ops.txt ${require_shape_fn} > ${GENERATE_PYTHON_OP_LIB_DESTINATION} DEPENDS ${tf_python_op_lib_name}_gen_python ) @@ -496,6 +512,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_tensor_forest_ops" + DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/contrib/tensor_forest/python/ops/gen_tensor_forest_ops.py) add_custom_target(tf_python_ops SOURCES ${tf_python_ops_generated_files} ${PYTHON_PROTO_GENFILES}) add_dependencies(tf_python_ops tf_python_op_gen_main) |