Fix include path in tpu_library_init_fns.inc for open source builds

PiperOrigin-RevId: 328559106 Change-Id: I986b7c2ef80cee0a2b40727aabd5491c42b155bd
2020-08-26 10:31:59 -07:00 · 2020-08-26 10:31:59 -07:00 · 25d003fc20
commit 25d003fc20
parent 42b4a44757
2 changed files with 5 additions and 0 deletions
--- a/tensorflow/core/tpu/BUILD
+++ b/tensorflow/core/tpu/BUILD
@ -157,6 +157,7 @@ cc_library(
        ":tpu_api",
        ":tpu_compilation_device",
        ":tpu_config_c_api",
+        ":tpu_executor_init_fns",
        ":tpu_library_init_fns",
        ":tpu_node_device",
        ":tpu_system_device",
--- a/tensorflow/core/tpu/tpu_library_init_fns.inc
+++ b/tensorflow/core/tpu/tpu_library_init_fns.inc
@ -1,4 +1,8 @@
+#if defined(PLATFORM_GOOGLE)
 #include "third_party/tensorflow/core/tpu/tpu_executor_init_fns.inc"
+#else
+#include "tensorflow/core/tpu/tpu_executor_init_fns.inc"
+#endif

 namespace {