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