diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index 1c9a37bf652..622480102a9 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -252,6 +252,7 @@ headers = ( list(find_files('*.h', 'tensorflow/c')) + list(find_files('*.h', 'tensorflow/cc')) + list(find_files('*.h', 'tensorflow/compiler')) + + list(find_files('*.h.inc', 'tensorflow/compiler')) + list(find_files('*.h', 'tensorflow/core')) + list(find_files('*.h', 'tensorflow/python')) + list(find_files('*.h', 'tensorflow/stream_executor')) +