[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:
parent
665cd8d46b
commit
d3f4283554
@ -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.
|
||||
|
@ -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')) +
|
||||
|
Loading…
Reference in New Issue
Block a user