experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #38328 from perfinion:systemlibs-pybind 

PiperOrigin-RevId: 305598649
Change-Id: I0a7845b01a7d0cb078d72e481ce640fc84977e9e
This commit is contained in:
TensorFlower Gardener 2020-04-08 18:48:06 -07:00
parent d5f5640cd9 dc86bb5296
commit 0277e80375
64 changed files with 125 additions and 114 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2tensorrt/utils/py_utils_wrapper.cc
View File

@ -15,7 +15,7 @@ limitations under the License.
#include <tuple> #include <tuple>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/compiler/tf2tensorrt/utils/py_utils.h" #include "tensorflow/compiler/tf2tensorrt/utils/py_utils.h"
std::tuple<int, int, int> get_linked_tensorrt_version() { std::tuple<int, int, int> get_linked_tensorrt_version() {

2
tensorflow/compiler/xla/python/bfloat16.h
View File

@ -16,7 +16,7 @@ limitations under the License.
#ifndef TENSORFLOW_COMPILER_XLA_PYTHON_BFLOAT16_H_ #ifndef TENSORFLOW_COMPILER_XLA_PYTHON_BFLOAT16_H_
#define TENSORFLOW_COMPILER_XLA_PYTHON_BFLOAT16_H_ #define TENSORFLOW_COMPILER_XLA_PYTHON_BFLOAT16_H_
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/compiler/xla/statusor.h" #include "tensorflow/compiler/xla/statusor.h"
namespace xla { namespace xla {

2
tensorflow/compiler/xla/python/dlpack.h
View File

@ -16,7 +16,7 @@ limitations under the License.
#ifndef TENSORFLOW_COMPILER_XLA_PYTHON_DLPACK_H_ #ifndef TENSORFLOW_COMPILER_XLA_PYTHON_DLPACK_H_
#define TENSORFLOW_COMPILER_XLA_PYTHON_DLPACK_H_ #define TENSORFLOW_COMPILER_XLA_PYTHON_DLPACK_H_
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/compiler/xla/python/local_client.h" #include "tensorflow/compiler/xla/python/local_client.h"
namespace xla { namespace xla {

2
tensorflow/compiler/xla/python/python_ref_manager.h
View File

@ -22,7 +22,7 @@ limitations under the License.
#include "absl/container/inlined_vector.h" #include "absl/container/inlined_vector.h"
#include "absl/synchronization/mutex.h" #include "absl/synchronization/mutex.h"
#include "absl/types/span.h" #include "absl/types/span.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
namespace xla { namespace xla {

2
tensorflow/compiler/xla/python/tpu_driver/client/tpu_client_extension.cc
View File

@ -15,7 +15,7 @@ limitations under the License.
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/compiler/xla/python/python_ref_manager.h" #include "tensorflow/compiler/xla/python/python_ref_manager.h"
#include "tensorflow/compiler/xla/python/tpu_driver/client/tpu_client.h" #include "tensorflow/compiler/xla/python/tpu_driver/client/tpu_client.h"
#include "tensorflow/compiler/xla/python/types.h" #include "tensorflow/compiler/xla/python/types.h"

6
tensorflow/compiler/xla/python/types.h
View File

@ -22,9 +22,9 @@ limitations under the License.
#include "numpy/arrayobject.h" #include "numpy/arrayobject.h"
#include "absl/container/inlined_vector.h" #include "absl/container/inlined_vector.h"
#include "absl/types/optional.h" #include "absl/types/optional.h"
#include "include/pybind11/numpy.h" #include "pybind11/numpy.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/compiler/xla/literal.h" #include "tensorflow/compiler/xla/literal.h"
#include "tensorflow/compiler/xla/python/local_client.h" #include "tensorflow/compiler/xla/python/local_client.h"
#include "tensorflow/compiler/xla/shape.h" #include "tensorflow/compiler/xla/shape.h"

8
tensorflow/compiler/xla/python/xla.cc
View File

@ -24,10 +24,10 @@ limitations under the License.
#include "absl/synchronization/mutex.h" #include "absl/synchronization/mutex.h"
#include "absl/types/optional.h" #include "absl/types/optional.h"
#include "absl/types/span.h" #include "absl/types/span.h"
#include "include/pybind11/cast.h" #include "pybind11/cast.h"
#include "include/pybind11/numpy.h" #include "pybind11/numpy.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/compiler/xla/client/client_library.h" #include "tensorflow/compiler/xla/client/client_library.h"
#include "tensorflow/compiler/xla/client/lib/comparators.h" #include "tensorflow/compiler/xla/client/lib/comparators.h"
#include "tensorflow/compiler/xla/client/lib/math.h" #include "tensorflow/compiler/xla/client/lib/math.h"

8
tensorflow/lite/experimental/support/codegen/python/codegen_lib.cc
View File

@ -13,10 +13,10 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/detail/common.h" #include "pybind11/detail/common.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/lite/experimental/support/codegen/android_java_generator.h" #include "tensorflow/lite/experimental/support/codegen/android_java_generator.h"
#include "tensorflow/lite/experimental/support/codegen/code_generator.h" #include "tensorflow/lite/experimental/support/codegen/code_generator.h"

6
tensorflow/lite/experimental/support/metadata/flatbuffers_lib/flatbuffers_lib.cc
View File

@ -15,9 +15,9 @@ limitations under the License.
#include "flatbuffers/flatbuffers.h" // from @flatbuffers #include "flatbuffers/flatbuffers.h" // from @flatbuffers
#include "flatbuffers/idl.h" // from @flatbuffers #include "flatbuffers/idl.h" // from @flatbuffers
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
namespace tflite { namespace tflite {
namespace support { namespace support {

6
tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper_pybind11.cc
View File

@ -13,9 +13,9 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.h" #include "tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

4
tensorflow/lite/python/optimize/calibration_wrapper_pybind11.cc
View File

@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/lite/python/optimize/calibration_wrapper.h" #include "tensorflow/lite/python/optimize/calibration_wrapper.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

4
tensorflow/lite/python/optimize/sparsification_wrapper_pybind11.cc
View File

@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/lite/python/optimize/sparsification_wrapper.h" #include "tensorflow/lite/python/optimize/sparsification_wrapper.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

4
tensorflow/lite/python/testdata/test_registerer_wrapper.cc vendored
View File

@ -9,8 +9,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/lite/python/testdata/test_registerer.h" #include "tensorflow/lite/python/testdata/test_registerer.h"
PYBIND11_MODULE(_pywrap_test_registerer, m) { PYBIND11_MODULE(_pywrap_test_registerer, m) {

4
tensorflow/lite/testing/string_util_wrapper.cc
View File

@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/lite/testing/string_util.h" #include "tensorflow/lite/testing/string_util.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

4
tensorflow/lite/tools/optimize/sparsity/python/format_converter_extension.cc
View File

@ -15,8 +15,8 @@ limitations under the License.
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/lite/c/common.h" #include "tensorflow/lite/c/common.h"
#include "tensorflow/lite/tools/optimize/sparsity/format_converter.h" #include "tensorflow/lite/tools/optimize/sparsity/format_converter.h"

6
tensorflow/python/autograph/impl/testing/pybind_for_testing.cc
View File

@ -13,9 +13,9 @@
// limitations under the License. // limitations under the License.
// ============================================================================== // ==============================================================================
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
namespace autograph { namespace autograph {

4
tensorflow/python/client/debug_events_writer_wrapper.cc
View File

@ -14,8 +14,8 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "absl/strings/string_view.h" #include "absl/strings/string_view.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/lib/core/status.h"
#include "tensorflow/core/lib/strings/stringprintf.h" #include "tensorflow/core/lib/strings/stringprintf.h"
#include "tensorflow/core/util/debug_events_writer.h" #include "tensorflow/core/util/debug_events_writer.h"

2
tensorflow/python/client/device_lib_wrapper.cc
View File

@ -15,7 +15,7 @@ limitations under the License.
#include <string> #include <string>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/device.h"
#include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/device_factory.h"
#include "tensorflow/core/framework/device_attributes.pb.h" #include "tensorflow/core/framework/device_attributes.pb.h"

4
tensorflow/python/client/events_writer_wrapper.cc
View File

@ -14,8 +14,8 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "absl/strings/string_view.h" #include "absl/strings/string_view.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/core/util/events_writer.h" #include "tensorflow/core/util/events_writer.h"
#include "tensorflow/python/lib/core/pybind11_absl.h" #include "tensorflow/python/lib/core/pybind11_absl.h"
#include "tensorflow/python/lib/core/pybind11_proto.h" #include "tensorflow/python/lib/core/pybind11_proto.h"

10
tensorflow/python/client/tf_session_wrapper.cc
View File

@ -15,11 +15,11 @@ limitations under the License.
#include "Python.h" #include "Python.h"
#include "absl/types/optional.h" #include "absl/types/optional.h"
#include "include/pybind11/chrono.h" #include "pybind11/chrono.h"
#include "include/pybind11/complex.h" #include "pybind11/complex.h"
#include "include/pybind11/functional.h" #include "pybind11/functional.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api.h"
#include "tensorflow/c/c_api_experimental.h" #include "tensorflow/c/c_api_experimental.h"
#include "tensorflow/c/c_api_internal.h" #include "tensorflow/c/c_api_internal.h"

12
tensorflow/python/data/service/server_lib_wrapper.cc
View File

@ -14,12 +14,12 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "Python.h" #include "Python.h"
#include "include/pybind11/chrono.h" #include "pybind11/chrono.h"
#include "include/pybind11/complex.h" #include "pybind11/complex.h"
#include "include/pybind11/functional.h" #include "pybind11/functional.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/core/data/service/server_lib.h" #include "tensorflow/core/data/service/server_lib.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"
#include "tensorflow/python/lib/core/pybind11_status.h" #include "tensorflow/python/lib/core/pybind11_status.h"

4
tensorflow/python/framework/dtypes.cc
View File

@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/detail/common.h" #include "pybind11/detail/common.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/framework/types.h" #include "tensorflow/core/framework/types.h"
#include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/framework/types.pb.h"

2
tensorflow/python/framework/memory_checker_test_helper.cc
View File

@ -15,7 +15,7 @@ limitations under the License.
#include <list> #include <list>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
namespace py = pybind11; namespace py = pybind11;

2
tensorflow/python/framework/op_def_registry.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op.h"
#include "tensorflow/core/framework/op_def.pb.h" #include "tensorflow/core/framework/op_def.pb.h"
#include "tensorflow/core/framework/op_def_util.h" #include "tensorflow/core/framework/op_def_util.h"

2
tensorflow/python/framework/python_memory_checker_helper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
namespace py = pybind11; namespace py = pybind11;

2
tensorflow/python/framework/python_op_gen_wrapper.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#include <Python.h> #include <Python.h>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/python/framework/python_op_gen.h" #include "tensorflow/python/framework/python_op_gen.h"
namespace py = pybind11; namespace py = pybind11;

4
tensorflow/python/grappler/cluster_wrapper.cc
View File

@ -24,8 +24,8 @@ limitations under the License.
#include <unordered_map> #include <unordered_map>
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/core/framework/kernel_def.pb.h" #include "tensorflow/core/framework/kernel_def.pb.h"
#include "tensorflow/core/framework/memory_types.h" #include "tensorflow/core/framework/memory_types.h"
#include "tensorflow/core/framework/op_def.pb.h" #include "tensorflow/core/framework/op_def.pb.h"

2
tensorflow/python/grappler/cost_analyzer_wrapper.cc
View File

@ -17,7 +17,7 @@ limitations under the License.
#include <sstream> #include <sstream>
#include <string> #include <string>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/grappler/clusters/single_machine.h" #include "tensorflow/core/grappler/clusters/single_machine.h"
#include "tensorflow/core/grappler/grappler_item.h" #include "tensorflow/core/grappler/grappler_item.h"
#include "tensorflow/core/grappler/grappler_item_builder.h" #include "tensorflow/core/grappler/grappler_item_builder.h"

2
tensorflow/python/grappler/graph_analyzer_tool_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/grappler/graph_analyzer/graph_analyzer_tool.h" #include "tensorflow/core/grappler/graph_analyzer/graph_analyzer_tool.h"
PYBIND11_MODULE(_pywrap_graph_analyzer_tool, m) { PYBIND11_MODULE(_pywrap_graph_analyzer_tool, m) {

4
tensorflow/python/grappler/item_wrapper.cc
View File

@ -19,8 +19,8 @@ limitations under the License.
#include <unordered_map> #include <unordered_map>
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/graph.pb.h"
#include "tensorflow/core/framework/node_def_util.h" #include "tensorflow/core/framework/node_def_util.h"
#include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op.h"

2
tensorflow/python/grappler/model_analyzer_wrapper.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#include <memory> #include <memory>
#include <sstream> #include <sstream>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/grappler/grappler_item_builder.h" #include "tensorflow/core/grappler/grappler_item_builder.h"
#include "tensorflow/core/protobuf/meta_graph.pb.h" #include "tensorflow/core/protobuf/meta_graph.pb.h"
#include "tensorflow/python/grappler/model_analyzer.h" #include "tensorflow/python/grappler/model_analyzer.h"

2
tensorflow/python/grappler/tf_optimizer_wrapper.cc
View File

@ -18,7 +18,7 @@ limitations under the License.
#include <string> #include <string>
#include <unordered_map> #include <unordered_map>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/device.h"
#include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/device_factory.h"
#include "tensorflow/core/framework/device_attributes.pb.h" #include "tensorflow/core/framework/device_attributes.pb.h"

2
tensorflow/python/lib/core/bfloat16_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/python/lib/core/bfloat16.h" #include "tensorflow/python/lib/core/bfloat16.h"
PYBIND11_MODULE(_pywrap_bfloat16, m) { PYBIND11_MODULE(_pywrap_bfloat16, m) {

4
tensorflow/python/lib/core/py_exception_registry_wrapper.cc
View File

@ -17,8 +17,8 @@ limitations under the License.
#include <array> #include <array>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/python/lib/core/py_exception_registry.h" #include "tensorflow/python/lib/core/py_exception_registry.h"
namespace py = pybind11; namespace py = pybind11;

2
tensorflow/python/lib/core/py_func_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/python/lib/core/py_func.h" #include "tensorflow/python/lib/core/py_func.h"
namespace py = pybind11; namespace py = pybind11;

2
tensorflow/python/lib/core/pybind11_absl.h
View File

@ -16,7 +16,7 @@ limitations under the License.
#ifndef TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_ABSL_H_ #ifndef TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_ABSL_H_
#define TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_ABSL_H_ #define TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_ABSL_H_
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/platform/stringpiece.h" #include "tensorflow/core/platform/stringpiece.h"
#ifndef ABSL_USES_STD_STRING_VIEW #ifndef ABSL_USES_STD_STRING_VIEW

4
tensorflow/python/lib/core/pybind11_lib.h
View File

@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#ifndef TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_LIB_H_ #ifndef TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_LIB_H_
#define TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_LIB_H_ #define TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_LIB_H_

2
tensorflow/python/lib/core/pybind11_proto.h
View File

@ -17,7 +17,7 @@ limitations under the License.
#define TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_PROTO_H_ #define TENSORFLOW_PYTHON_LIB_CORE_PYBIND11_PROTO_H_
#include "absl/strings/str_cat.h" #include "absl/strings/str_cat.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
namespace tensorflow { namespace tensorflow {

2
tensorflow/python/lib/core/pybind11_status.h
View File

@ -18,7 +18,7 @@ limitations under the License.
#include <Python.h> #include <Python.h>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status.h"
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/lib/core/status.h"
#include "tensorflow/core/protobuf/error_codes.pb.h" #include "tensorflow/core/protobuf/error_codes.pb.h"

4
tensorflow/python/lib/io/file_io_wrapper.cc
View File

@ -17,8 +17,8 @@ limitations under the License.
#include <string> #include <string>
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/core/lib/core/error_codes.pb.h" #include "tensorflow/core/lib/core/error_codes.pb.h"
#include "tensorflow/core/lib/core/errors.h" #include "tensorflow/core/lib/core/errors.h"
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/lib/core/status.h"

2
tensorflow/python/lib/io/record_io_wrapper.cc
View File

@ -17,7 +17,7 @@ limitations under the License.
#include <string> #include <string>
#include "absl/memory/memory.h" #include "absl/memory/memory.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/lib/core/errors.h" #include "tensorflow/core/lib/core/errors.h"
#include "tensorflow/core/lib/core/stringpiece.h" #include "tensorflow/core/lib/core/stringpiece.h"
#include "tensorflow/core/lib/io/record_reader.h" #include "tensorflow/core/lib/io/record_reader.h"

2
tensorflow/python/lite/toco_python_api_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/lite/toco/python/toco_python_api.h" #include "tensorflow/lite/toco/python/toco_python_api.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

2
tensorflow/python/mlir_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status.h"
#include "tensorflow/compiler/mlir/python/mlir.h" #include "tensorflow/compiler/mlir/python/mlir.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

2
tensorflow/python/platform/stacktrace_handler_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/platform/stacktrace_handler.h" #include "tensorflow/core/platform/stacktrace_handler.h"
namespace py = pybind11; namespace py = pybind11;

4
tensorflow/python/profiler/internal/profiler_wrapper.cc
View File

@ -16,8 +16,8 @@ limitations under the License.
#include <memory> #include <memory>
#include "absl/memory/memory.h" #include "absl/memory/memory.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/core/platform/host_info.h" #include "tensorflow/core/platform/host_info.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"
#include "tensorflow/core/profiler/convert/xplane_to_profile_response.h" #include "tensorflow/core/profiler/convert/xplane_to_profile_response.h"

6
tensorflow/python/profiler/internal/python_hooks.h
View File

@ -19,9 +19,9 @@ limitations under the License.
#include <vector> #include <vector>
#include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_map.h"
#include "include/pybind11/cast.h" #include "pybind11/cast.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"
#include "tensorflow/core/profiler/lib/traceme.h" #include "tensorflow/core/profiler/lib/traceme.h"

2
tensorflow/python/profiler/internal/scoped_annotation_wrapper.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#include <utility> #include <utility>
#include "absl/types/optional.h" #include "absl/types/optional.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"
#include "tensorflow/core/profiler/lib/scoped_annotation.h" #include "tensorflow/core/profiler/lib/scoped_annotation.h"

2
tensorflow/python/profiler/internal/traceme_wrapper.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#include <utility> #include <utility>
#include "absl/types/optional.h" #include "absl/types/optional.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"
#include "tensorflow/core/profiler/lib/traceme.h" #include "tensorflow/core/profiler/lib/traceme.h"

2
tensorflow/python/pywrap_tensorflow_internal.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
// This logic allows Python to import _pywrap_tensorflow_internal.so by // This logic allows Python to import _pywrap_tensorflow_internal.so by
// creating a PyInit function and exposing it. It is required in opensource // creating a PyInit function and exposing it. It is required in opensource

8
tensorflow/python/tfcompile_wrapper.cc
View File

@ -15,10 +15,10 @@ limitations under the License.
#include <string> #include <string>
#include "include/pybind11/cast.h" #include "pybind11/cast.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/compiler/aot/compile.h" #include "tensorflow/compiler/aot/compile.h"
#include "tensorflow/compiler/aot/flags.h" #include "tensorflow/compiler/aot/flags.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"

10
tensorflow/python/tfe_wrapper.cc
View File

@ -16,11 +16,11 @@ limitations under the License.
#include <memory> #include <memory>
#include "Python.h" #include "Python.h"
#include "include/pybind11/chrono.h" #include "pybind11/chrono.h"
#include "include/pybind11/complex.h" #include "pybind11/complex.h"
#include "include/pybind11/functional.h" #include "pybind11/functional.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api.h"
#include "tensorflow/c/c_api_experimental.h" #include "tensorflow/c/c_api_experimental.h"
#include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api.h"

2
tensorflow/python/training/quantize_training_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/graph/quantize_training.h" #include "tensorflow/core/graph/quantize_training.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"
#include "tensorflow/python/lib/core/pybind11_status.h" #include "tensorflow/python/lib/core/pybind11_status.h"

2
tensorflow/python/util/kernel_registry_wrapper.cc
View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/python/util/kernel_registry.h" #include "tensorflow/python/util/kernel_registry.h"
namespace py = pybind11; namespace py = pybind11;

4
tensorflow/python/util/port_wrapper.cc
View File

@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/core/util/port.h" #include "tensorflow/core/util/port.h"
PYBIND11_MODULE(_pywrap_util_port, m) { PYBIND11_MODULE(_pywrap_util_port, m) {

10
tensorflow/python/util/py_checkpoint_reader_wrapper.cc
View File

@ -18,11 +18,11 @@ limitations under the License.
#include "numpy/arrayobject.h" #include "numpy/arrayobject.h"
#include "numpy/ufuncobject.h" #include "numpy/ufuncobject.h"
#include "include/pybind11/chrono.h" #include "pybind11/chrono.h"
#include "include/pybind11/complex.h" #include "pybind11/complex.h"
#include "include/pybind11/functional.h" #include "pybind11/functional.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl.h" #include "pybind11/stl.h"
#include "tensorflow/c/checkpoint_reader.h" #include "tensorflow/c/checkpoint_reader.h"
#include "tensorflow/c/tf_status.h" #include "tensorflow/c/tf_status.h"
#include "tensorflow/core/lib/core/errors.h" #include "tensorflow/core/lib/core/errors.h"

4
tensorflow/python/util/stat_summarizer_wrapper.cc
View File

@ -16,8 +16,8 @@ limitations under the License.
#include <memory> #include <memory>
#include "absl/memory/memory.h" #include "absl/memory/memory.h"
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/graph.pb.h"
#include "tensorflow/core/framework/step_stats.pb.h" #include "tensorflow/core/framework/step_stats.pb.h"
#include "tensorflow/core/util/stat_summarizer.h" #include "tensorflow/core/util/stat_summarizer.h"

4
tensorflow/python/util/tf_stack.cc
View File

@ -19,8 +19,8 @@ limitations under the License.
#include <algorithm> #include <algorithm>
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/stl_bind.h" #include "pybind11/stl_bind.h"
struct FrameSummary; // Forward declaration. struct FrameSummary; // Forward declaration.

2
tensorflow/python/util/tfprof_wrapper.cc
View File

@ -15,7 +15,7 @@ limitations under the License.
#include <string> #include <string>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/framework/types.h" #include "tensorflow/core/framework/types.h"
#include "tensorflow/core/profiler/internal/print_model_analysis.h" #include "tensorflow/core/profiler/internal/print_model_analysis.h"

2
tensorflow/python/util/transform_graph_wrapper.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#include <string> #include <string>
#include <vector> #include <vector>
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "tensorflow/core/lib/strings/str_util.h" #include "tensorflow/core/lib/strings/str_util.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"
#include "tensorflow/python/lib/core/pybind11_status.h" #include "tensorflow/python/lib/core/pybind11_status.h"

4
tensorflow/python/util/util_wrapper.cc
View File

@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "include/pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "include/pybind11/pytypes.h" #include "pybind11/pytypes.h"
#include "tensorflow/python/lib/core/pybind11_lib.h" #include "tensorflow/python/lib/core/pybind11_lib.h"
#include "tensorflow/python/util/util.h" #include "tensorflow/python/util/util.h"

1
tensorflow/workspace.bzl
View File

@ -1039,6 +1039,7 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
sha256 = "1eed57bc6863190e35637290f97a20c81cfe4d9090ac0a24f3bbf08f265eb71d", sha256 = "1eed57bc6863190e35637290f97a20c81cfe4d9090ac0a24f3bbf08f265eb71d",
strip_prefix = "pybind11-2.4.3", strip_prefix = "pybind11-2.4.3",
build_file = clean_dep("//third_party:pybind11.BUILD"), build_file = clean_dep("//third_party:pybind11.BUILD"),
system_build_file = clean_dep("//third_party/systemlibs:pybind11.BUILD"),
) )
tf_http_archive( tf_http_archive(

1
third_party/pybind11.BUILD vendored
View File

@ -18,6 +18,7 @@ cc_library(
"-Wno-pragma-once-outside-header", "-Wno-pragma-once-outside-header",
], ],
includes = ["include"], includes = ["include"],
strip_include_prefix = "include",
deps = [ deps = [
"@org_tensorflow//third_party/python_runtime:headers", "@org_tensorflow//third_party/python_runtime:headers",
], ],

8
third_party/systemlibs/pybind11.BUILD vendored Normal file
View File

@ -0,0 +1,8 @@
package(default_visibility = ["//visibility:public"])
cc_library(
name = "pybind11",
deps = [
"@org_tensorflow//third_party/python_runtime:headers",
],
)

1
third_party/systemlibs/syslibs_configure.bzl vendored
View File

@ -38,6 +38,7 @@ VALID_LIBS = [
"pasta", "pasta",
"pcre", "pcre",
"png", "png",
"pybind11",
"six_archive", "six_archive",
"snappy", "snappy",
"swig", "swig",