diff options
Diffstat (limited to 'tensorflow/contrib/cmake')
-rwxr-xr-x | tensorflow/contrib/cmake/tf_python.cmake | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_python.cmake b/tensorflow/contrib/cmake/tf_python.cmake index e3b59001bc..8a9172b43c 100755 --- a/tensorflow/contrib/cmake/tf_python.cmake +++ b/tensorflow/contrib/cmake/tf_python.cmake @@ -853,6 +853,7 @@ add_custom_command( "--apidir=${CMAKE_CURRENT_BINARY_DIR}/tf_python/tensorflow/python/estimator/api" "--package=tensorflow.python.estimator" "--apiname=estimator" + "--output_package=tensorflow.python.estimator.api" "${estimator_api_init_list_file}" COMMENT "Generating __init__.py files for Python API." |