Skip to content

Commit 68ff6a2

Browse files
committed
Create missing API defs on generation
+ generate defs native artifact to contains TF native resources only required at generation
1 parent d85b300 commit 68ff6a2

File tree

2,960 files changed

+2818
-15038
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

2,960 files changed

+2818
-15038
lines changed

.github/workflows/build.yml

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ env:
1616
jobs:
1717
check-format:
1818
if: github.event_name == 'pull_request'
19-
runs-on: ubuntu-18.04
19+
runs-on: ubuntu-20.04
2020
steps:
2121
- name: Configure Java
2222
uses: actions/setup-java@v2
@@ -50,7 +50,6 @@ jobs:
5050
echo "Staging repository created: $STAGING_REPOSITORY_ID"
5151
echo "::set-output name=stagingRepositoryId::$STAGING_REPOSITORY_ID"
5252
linux-x86_64:
53-
if: github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'CI build')
5453
runs-on: ubuntu-20.04
5554
needs: prepare
5655
strategy:
@@ -71,7 +70,6 @@ jobs:
7170
echo "<settings><servers><server><id>ossrh</id><username>${{ secrets.CI_DEPLOY_USERNAME }}</username><password>${{ secrets.CI_DEPLOY_PASSWORD }}</password></server></servers></settings>" > $HOME/.m2/settings.xml
7271
mvn clean install -pl '!tensorflow-framework' -B -U -e -Djavacpp.platform=linux-x86_64 -Djavacpp.platform.extension=${{ matrix.ext }} -am -DstagingRepositoryId=${{ needs.prepare.outputs.stagingRepositoryId }}
7372
macosx-x86_64:
74-
if: github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'CI build')
7573
runs-on: macos-11
7674
needs: prepare
7775
strategy:
@@ -131,7 +129,7 @@ jobs:
131129
call mvn -version
132130
echo ^<settings^>^<servers^>^<server^>^<id^>ossrh^</id^>^<username^>${{ secrets.CI_DEPLOY_USERNAME }}^</username^>^<password^>${{ secrets.CI_DEPLOY_PASSWORD }}^</password^>^</server^>^</servers^>^</settings^> > %USERPROFILE%\.m2\settings.xml
133131
set "SKIP_EXPORT=true"
134-
call mvn clean install -pl "!tensorflow-framework" -B -U -e -Djavacpp.platform=windows-x86_64 -Djavacpp.platform.extension=${{ matrix.ext }} -am -DstagingRepositoryId=${{ needs.prepare.outputs.stagingRepositoryId }}
132+
call mvn clean install -pl "!tensorflow-framework" -B -U -e -Dnative.build -Djavacpp.platform=windows-x86_64 -Djavacpp.platform.extension=${{ matrix.ext }} -am -DstagingRepositoryId=${{ needs.prepare.outputs.stagingRepositoryId }}
135133
if ERRORLEVEL 1 exit /b
136134
deploy:
137135
if: github.event_name == 'push' && contains(github.ref, 'master')

tensorflow-core/tensorflow-core-api/pom.xml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@
107107
<argument>${project.basedir}/src/api</argument>
108108
<argument>-o</argument>
109109
<argument>${project.basedir}/src/gen/java</argument>
110+
<argument>-c</argument>
110111
</arguments>
111112
</configuration>
112113
</execution>
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
op {
2+
graph_op_name: "CollectiveAllToAllV2"
3+
visibility: SKIP
4+
}
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "CollectiveReduceScatterV2"
3+
endpoint {
4+
name: "collective.CollectiveReduceScatter"
5+
}
6+
}
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "ComputeDedupDataTupleMask"
3+
endpoint {
4+
name: "tpu.ComputeDedupDataTupleMask"
5+
}
6+
}
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "Conv"
3+
endpoint {
4+
name: "nn.Conv"
5+
}
6+
}
Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
11
op {
22
graph_op_name: "Conv2DBackpropFilter"
3-
endpoint {
4-
name: "nn.Conv2dBackpropFilter"
5-
}
3+
visibility: SKIP
64
}
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
op {
2+
graph_op_name: "Conv2DBackpropFilterV2"
3+
visibility: VISIBLE
4+
endpoint {
5+
name: "nn.Conv2DBackpropFilter"
6+
}
7+
}
Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
11
op {
22
graph_op_name: "Conv2DBackpropInput"
3-
endpoint {
4-
name: "nn.Conv2dBackpropInput"
5-
}
3+
visibility: SKIP
64
}
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
op {
2+
graph_op_name: "Conv2DBackpropInputV2"
3+
visibility: VISIBLE
4+
endpoint {
5+
name: "nn.Conv2DBackpropInput"
6+
}
7+
}

0 commit comments

Comments
 (0)