diff --git a/tensorflow-core/tensorflow-core-api/build.sh b/tensorflow-core/tensorflow-core-api/build.sh
index b1c5194214f..895832f3df1 100755
--- a/tensorflow-core/tensorflow-core-api/build.sh
+++ b/tensorflow-core/tensorflow-core-api/build.sh
@@ -79,15 +79,18 @@ fi
GEN_SRCS_DIR=src/gen/java
mkdir -p $GEN_SRCS_DIR
-GEN_RESOURCE_DIR=src/gen/resources/org/tensorflow/op
+GEN_RESOURCE_DIR=src/gen/resources
mkdir -p $GEN_RESOURCE_DIR
if [[ -z "${SKIP_EXPORT:-}" ]]; then
# Export op defs
echo "Exporting Ops"
$BAZEL_BIN/java_op_exporter \
- --api_dirs=$BAZEL_SRCS/external/org_tensorflow/tensorflow/core/api_def/base_api,src/bazel/api_def \
- $TENSORFLOW_LIB > $GEN_RESOURCE_DIR/ops.pb
+ $TENSORFLOW_LIB \
+ $GEN_RESOURCE_DIR/ops.pb \
+ $GEN_RESOURCE_DIR/ops.pbtxt \
+ $BAZEL_SRCS/external/org_tensorflow/tensorflow/core/api_def/base_api \
+ src/bazel/api_def
else
echo "Skipping Op export"
fi
diff --git a/tensorflow-core/tensorflow-core-api/pom.xml b/tensorflow-core/tensorflow-core-api/pom.xml
index ecbb3cca67c..9e5df6de251 100644
--- a/tensorflow-core/tensorflow-core-api/pom.xml
+++ b/tensorflow-core/tensorflow-core-api/pom.xml
@@ -45,15 +45,15 @@
ndarray
${project.version}
-
- org.junit.jupiter
- junit-jupiter-api
- test
-
-
- org.junit.jupiter
- junit-jupiter-engine
- test
+
+ org.junit.jupiter
+ junit-jupiter-api
+ test
+
+
+ org.junit.jupiter
+ junit-jupiter-engine
+ test
org.openjdk.jmh
@@ -359,7 +359,7 @@
org.tensorflow.op.generator.OpGenerator
${project.basedir}/src/gen/java
- ${project.basedir}/src/gen/resources/org/tensorflow/op/ops.pb
+ ${project.basedir}/src/gen/resources/ops.pb
@@ -423,10 +423,10 @@
-
-
+
+
${project.build.directory}/${project.artifactId}-${project.version}-${native.classifier}.jar