ErrorCodes proto fix.

tensorflow · Jan 12, 2024 · 5aeda89 · 5aeda89
1 parent d1b1168
commit 5aeda89
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 3 deletions.
diff --git a/tensorflow-core/tensorflow-core-native/WORKSPACE b/tensorflow-core/tensorflow-core-native/WORKSPACE
@@ -9,6 +9,7 @@ http_archive(
     name = "org_tensorflow",
     patches = [
         ":tensorflow-visibility.patch",
+        ":tensorflow-proto.patch",
     ],
     patch_tool = "patch",
     patch_args = ["-p1"],

diff --git a/tensorflow-core/tensorflow-core-native/external/tensorflow-proto.patch b/tensorflow-core/tensorflow-core-native/external/tensorflow-proto.patch
@@ -0,0 +1,9 @@
+diff -ruN tensorflow/tensorflow/core/lib/core/error_codes.proto tensorflow-proto-patch/tensorflow/core/lib/core/error_codes.proto
+--- tensorflow/tensorflow/core/lib/core/error_codes.proto
++++ tensorflow-proto-patch/tensorflow/core/lib/core/error_codes.proto
+@@ -1,3 +1,5 @@
+ syntax = "proto3";
+
++package tensorflow;
++
+ import public "tsl/protobuf/error_codes.proto";
diff --git a/...-core-native/src/gen/java/ErrorCodes.java → ...java/org/tensorflow/proto/ErrorCodes.java b/...-core-native/src/gen/java/ErrorCodes.java → ...java/org/tensorflow/proto/ErrorCodes.java
diff --git a/tensorflow-core/tensorflow-core-native/src/main/java/module-info.java b/tensorflow-core/tensorflow-core-native/src/main/java/module-info.java
@@ -23,7 +23,7 @@
   exports org.tensorflow.internal;
   exports org.tensorflow.internal.c_api;
   exports org.tensorflow.internal.c_api.global;
-  exports org.tensorflow.internal.c_api.presets to org.bytedeco.javacpp;
+  exports org.tensorflow.internal.c_api.presets to org.bytedeco.javacpp,tensorflow;
   exports org.tensorflow.proto;
   exports org.tensorflow.proto.data;
   exports org.tensorflow.proto.data.model;