[TF pip] Include tensorflow/cc/saved_model/**.h in tensorflow pip includes.

TESTED:
* Built build_pip_package locally and verified the wheel contains the required
  files.
PiperOrigin-RevId: 300189399
Change-Id: If3ef8c6cbd39a24409cea5999c916b0f8a253713
This commit is contained in:
Eugene Brevdo 2020-03-10 15:19:38 -07:00 committed by TensorFlower Gardener
parent 665cd8d46b
commit d3f4283554
2 changed files with 4 additions and 0 deletions

View File

@ -33,6 +33,9 @@ transitive_hdrs(
"//tensorflow/core:lib",
"//tensorflow/core:protos_all_cc",
"//tensorflow/core:stream_executor",
"//tensorflow/cc/saved_model:loader",
"//tensorflow/cc/saved_model:reader",
"//tensorflow/cc/saved_model:bundle_v2",
# WARNING: None of the C/C++ code under python/ has any API guarantees, and TF team
# reserves the right to change APIs and other header-level interfaces. If your custom
# op uses these headers, it may break when users upgrade their version of tensorflow.

View File

@ -246,6 +246,7 @@ headers = (
list(find_files('*.proto', 'tensorflow/compiler')) +
list(find_files('*.proto', 'tensorflow/core')) +
list(find_files('*.proto', 'tensorflow/python')) +
list(find_files('*.h', 'tensorflow/cc')) +
list(find_files('*.h', 'tensorflow/compiler')) +
list(find_files('*.h', 'tensorflow/core')) +
list(find_files('*.h', 'tensorflow/python')) +