Fixed Serving server startup in docker environment

PiperOrigin-RevId: 896112371

Author: dmiltr3

WORKSPACE

@@ -32,6 +32,8 @@ tensorflow_http_archive(
         "echo -e \"\\ndef cc_library_oss(deps=[], **kwargs):\\n    if kwargs.get(\\\"name\\\") == \\\"lib_internal_impl\\\" or \\\"protobuf\\\" in kwargs.get(\\\"name\\\", \\\"\\\"):\\n        _cc_library(deps = deps, **kwargs)\\n        return\\n    if type(deps) == \\\"list\\\":\\n        if \\\"@com_google_protobuf//:protobuf\\\" not in deps:\\n            deps = deps + [\\\"@com_google_protobuf//:protobuf\\\"]\\n    else:\\n        deps = deps + [\\\"@com_google_protobuf//:protobuf\\\"]\\n    _cc_library(deps = deps, **kwargs)\\ncc_library = cc_library_oss\" >> tensorflow/core/platform/rules_cc.bzl",
         "sed -i 's#deps = \\[op_gen\\] + deps#deps = [op_gen] + deps + [clean_dep(\"//tensorflow/core/framework:kernel_shape_util\"), clean_dep(\"//tensorflow/core/framework:full_type_util\")]#' tensorflow/tensorflow.bzl",
         "sed -i '/name = \"kernel_shape_util\",/a \\    visibility = [\"//visibility:public\"],' tensorflow/core/framework/BUILD",
+        "echo -e '\\nalias(name = \"tensorflow_libtensorflow_framework\", actual = \"//tensorflow/core:tensorflow\", visibility = [\"//visibility:public\"])' >> BUILD",
+        "echo -e '\\nalias(name = \"tensorflow_tf_header_lib\", actual = \"//tensorflow/core:tensorflow\", visibility = [\"//visibility:public\"])' >> BUILD",
     ],
 )
 

third_party/tensorflow/BUILD

@@ -1,11 +1,11 @@
 alias(
     name = "tensorflow_libtensorflow_framework",
-    actual = "//tensorflow:libtensorflow_framework_import_lib",
+    actual = "//tensorflow/core:tensorflow",
     visibility = ["//visibility:public"],
 )
 
 alias(
     name = "tensorflow_tf_header_lib",
-    actual = "//tensorflow/core:framework_headers_lib",
+    actual = "//tensorflow/core:tensorflow",
     visibility = ["//visibility:public"],
 )

third_party/tensorflow/tensorflow.patch

@@ -1,5 +1,4 @@
 diff --git a/tensorflow/tools/toolchains/python/python_repo.bzl b/tensorflow/tools/toolchains/python/python_repo.bzl
-index 47fe64d7b7b..a01a1f19c8b 100644
 --- a/tensorflow/tools/toolchains/python/python_repo.bzl
 +++ b/tensorflow/tools/toolchains/python/python_repo.bzl
 @@ -21,5 +21,6 @@
@@ -9,38 +8,13 @@ index 47fe64d7b7b..a01a1f19c8b 100644
  WHEEL_COLLAB = "{}"
 +USE_PYWRAP_RULES = "False"
  """
- 
- def _python_repository_impl(repository_ctx):
 diff --git a/tensorflow/core/framework/tensor_testutil.cc b/tensorflow/core/framework/tensor_testutil.cc
 --- a/tensorflow/core/framework/tensor_testutil.cc
 +++ b/tensorflow/core/framework/tensor_testutil.cc
-@@ -16,6 +16,7 @@
+@@ -16,5 +16,6 @@
  #include "tensorflow/core/framework/tensor_testutil.h"
 
  #include <cmath>
 +#include <iomanip>
 
  #include "tensorflow/core/framework/tensor.h"
- #include "tensorflow/core/platform/types.h"
-diff --git a/BUILD b/BUILD
---- a/BUILD
-+++ b/BUILD
-@@ -3,3 +3,18 @@
-     "AUTHORS",
-     "LICENSE",
- ])
-+
-+cc_import(
-+    name = "libtensorflow_framework_import",
-+    shared_library = "//tensorflow:libtensorflow_framework.so",
-+)
-+alias(
-+    name = "tensorflow_libtensorflow_framework",
-+    actual = ":libtensorflow_framework_import",
-+    visibility = ["//visibility:public"],
-+)
-+alias(
-+    name = "tensorflow_tf_header_lib",
-+    actual = "//tensorflow/core:framework_headers_lib",
-+    visibility = ["//visibility:public"],
-+)