diff options
Diffstat (limited to 'tensorflow/tools')
| -rw-r--r-- | tensorflow/tools/pip_package/setup.py | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index d864a7a039..dd1dca9ee8 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -226,13 +226,14 @@ if os.name == 'nt': else: EXTENSION_NAME = 'python/_pywrap_tensorflow_internal.so' -headers = (list(find_files('*.h', 'tensorflow/core')) + - list(find_files('*.h', 'tensorflow/stream_executor')) + - list(find_files('*.h', 'google/protobuf_archive/src')) + - list(find_files('*', 'third_party/eigen3')) + - list(find_files('*.h', - 'tensorflow/include/external/com_google_absl')) + - list(find_files('*', 'tensorflow/include/external/eigen_archive'))) +headers = ( + list(find_files('*.h', 'tensorflow/core')) + list( + find_files('*.h', 'tensorflow/stream_executor')) + + list(find_files('*.h', 'google/protobuf_archive/src')) + list( + find_files('*', 'third_party/eigen3')) + list( + find_files('*.h', 'tensorflow/include/external/com_google_absl')) + + list(find_files('*.inc', 'tensorflow/include/external/com_google_absl')) + + list(find_files('*', 'tensorflow/include/external/eigen_archive'))) setup( name=project_name, |