tensorflow · karllessard · Jun 26, 2021 · May 31, 2021 · May 31, 2021 · May 31, 2021
diff --git a/pom.xml b/pom.xml
@@ -1,4 +1,6 @@
-<project xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd" xmlns="http://maven.apache.org/POM/4.0.0"
+<project
+  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"
+  xmlns="http://maven.apache.org/POM/4.0.0"
   xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
 
   <modelVersion>4.0.0</modelVersion>
@@ -86,7 +88,9 @@
     </snapshotRepository>
     <repository>
       <id>ossrh</id>
-      <url>https://oss.sonatype.org/service/local/staging/deployByRepositoryId/${stagingRepositoryId}/</url>
+      <url>
+        https://oss.sonatype.org/service/local/staging/deployByRepositoryId/${stagingRepositoryId}/
+      </url>
     </repository>
   </distributionManagement>
 
@@ -156,7 +160,9 @@
         <repository>
           <id>ossrh-staging</id>
           <name>OSSRH Sonatype Staging</name>
-          <url>https://oss.sonatype.org/service/local/staging/deployByRepositoryId/${stagingRepositoryId}/</url>
+          <url>
+            https://oss.sonatype.org/service/local/staging/deployByRepositoryId/${stagingRepositoryId}/
+          </url>
           <releases>
             <enabled>true</enabled>
           </releases>
@@ -230,6 +236,33 @@
       </build>
     </profile>
 
+    <!--
+      Profile to run spotless:apply on builds.  Will run before format's check.
+    -->
+    <profile>
+      <id>apply-format</id>
+      <build>
+        <plugins>
+          <plugin>
+            <groupId>com.diffplug.spotless</groupId>
+            <artifactId>spotless-maven-plugin</artifactId>
+            <version>${spotless.version}</version>
+
+            <executions>
+              <execution>
+                <!-- Runs in initialize phase to fail fast in case of formatting issues (should be before codegen).-->
+                <id>spotless-apply</id>
+                <phase>initialize</phase>
+                <goals>
+                  <goal>apply</goal>
+                </goals>
+              </execution>
+            </executions>
+          </plugin>
+        </plugins>
+      </build>
+    </profile>
+
     <!--
       Profile that enables format checks on compilation
       Can auto-format using spotless:apply.

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_AssertCardinalityDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_AssertCardinalityDataset.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "AssertCardinalityDataset"
+  visibility: VISIBLE
   endpoint {
-    name: "data.experimental.AssertCardinalityDataset"
+    name: "data.AssertCardinalityDataset"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_AssertNextDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_AssertNextDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "AssertNextDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.AssertNextDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_AutoShardDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_AutoShardDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "AutoShardDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.AutoShardDataset"
   }

diff --git a/...orflow-core/tensorflow-core-api/src/bazel/api_def/api_def_BytesProducedStatsDataset.pbtxt b/...orflow-core/tensorflow-core-api/src/bazel/api_def/api_def_BytesProducedStatsDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "BytesProducedStatsDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.BytesProducedStatsDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CSVDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CSVDataset.pbtxt
@@ -1,6 +1,4 @@
 op {
   graph_op_name: "CSVDataset"
-  endpoint {
-    name: "data.CSVDataset"
-  }
+  visibility: SKIP
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CSVDatasetV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CSVDatasetV2.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "CSVDatasetV2"
+  visibility: VISIBLE
   endpoint {
-    name: "data.CSVDatasetV2"
+    name: "data.CSVDataset"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CacheDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CacheDataset.pbtxt
@@ -1,6 +1,4 @@
 op {
   graph_op_name: "CacheDataset"
-  endpoint {
-    name: "data.CacheDataset"
-  }
+  visibility: SKIP
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CacheDatasetV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CacheDatasetV2.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "CacheDatasetV2"
+  visibility: VISIBLE
   endpoint {
-    name: "data.CacheDatasetV2"
+    name: "data.CacheDataset"
   }
 }
diff --git a/...rflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ChooseFastestBranchDataset.pbtxt b/...rflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ChooseFastestBranchDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "ChooseFastestBranchDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.ChooseFastestBranchDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ChooseFastestDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ChooseFastestDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "ChooseFastestDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.ChooseFastestDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CompressElement.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CompressElement.pbtxt
@@ -1,6 +1,6 @@
 op {
   graph_op_name: "CompressElement"
   endpoint {
-    name: "data.experimental.CompressElement"
+    name: "data.CompressElement"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DataServiceDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DataServiceDataset.pbtxt
@@ -1,6 +1,4 @@
 op {
   graph_op_name: "DataServiceDataset"
-  endpoint {
-    name: "data.experimental.DataServiceDataset"
-  }
+  visibility: SKIP
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DataServiceDatasetV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DataServiceDatasetV2.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "DataServiceDatasetV2"
+  visibility: VISIBLE
   endpoint {
-    name: "rawops.DataServiceDatasetV2"
+    name: "data.DataServiceDatasetV2"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetCardinality.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetCardinality.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DatasetCardinality"
+  visibility: VISIBLE
   endpoint {
     name: "data.DatasetCardinality"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetFromGraph.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetFromGraph.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DatasetFromGraph"
+  visibility: VISIBLE
   endpoint {
     name: "data.DatasetFromGraph"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetToGraphV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetToGraphV2.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DatasetToGraphV2"
+  visibility: VISIBLE
   endpoint {
     name: "data.DatasetToGraph"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetToSingleElement.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetToSingleElement.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DatasetToSingleElement"
+  visibility: VISIBLE
   endpoint {
     name: "data.DatasetToSingleElement"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetToTFRecord.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DatasetToTFRecord.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DatasetToTFRecord"
+  visibility: VISIBLE
   endpoint {
     name: "data.DatasetToTfRecord"
   }

diff --git a/...orflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DenseToSparseBatchDataset.pbtxt b/...orflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DenseToSparseBatchDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DenseToSparseBatchDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.DenseToSparseBatchDataset"
   }

diff --git a/...orflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DirectedInterleaveDataset.pbtxt b/...orflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DirectedInterleaveDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "DirectedInterleaveDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.DirectedInterleaveDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DummyIterationCounter.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DummyIterationCounter.pbtxt
@@ -1,6 +1,6 @@
 op {
   graph_op_name: "DummyIterationCounter"
   endpoint {
-    name: "data.experimental.DummyIterationCounter"
+    name: "data.DummyIterationCounter"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DummySeedGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DummySeedGenerator.pbtxt
@@ -1,6 +1,6 @@
 op {
   graph_op_name: "DummySeedGenerator"
   endpoint {
-    name: "random.experimental.DummySeedGenerator"
+    name: "random.DummySeedGenerator"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_EnqueueInQueueDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_EnqueueInQueueDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "EnqueueInQueueDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.EnqueueInQueueDataset"
   }

diff --git a/...low-core/tensorflow-core-api/src/bazel/api_def/api_def_FilterByLastComponentDataset.pbtxt b/...low-core/tensorflow-core-api/src/bazel/api_def/api_def_FilterByLastComponentDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "FilterByLastComponentDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.FilterByLastComponentDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_FinalizeDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_FinalizeDataset.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "FinalizeDataset"
+  visibility: VISIBLE
   endpoint {
-    name: "rawops.FinalizeDataset"
+    name: "data.FinalizeDataset"
   }
 }
diff --git a/...rflow-core/tensorflow-core-api/src/bazel/api_def/api_def_FixedLengthRecordDatasetV2.pbtxt b/...rflow-core/tensorflow-core-api/src/bazel/api_def/api_def_FixedLengthRecordDatasetV2.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "FixedLengthRecordDatasetV2"
+  visibility: VISIBLE
   endpoint {
     name: "data.FixedLengthRecordDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GeneratorDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GeneratorDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "GeneratorDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.GeneratorDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GroupByReducerDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GroupByReducerDataset.pbtxt
@@ -1,3 +1,7 @@
 op {
   graph_op_name: "GroupByReducerDataset"
+  visibility: VISIBLE
+  endpoint {
+    name: "data.GroupByReducerDataset"
+  }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GroupByWindowDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GroupByWindowDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "GroupByWindowDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.GroupByWindowDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_IgnoreErrorsDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_IgnoreErrorsDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "IgnoreErrorsDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.IgnoreErrorsDataset"
   }

diff --git a/...rflow-core/tensorflow-core-api/src/bazel/api_def/api_def_InitializeTableFromDataset.pbtxt b/...rflow-core/tensorflow-core-api/src/bazel/api_def/api_def_InitializeTableFromDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "InitializeTableFromDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.InitializeTableFromDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_KafkaDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_KafkaDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "KafkaDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.KafkaDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LMDBDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LMDBDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "LMDBDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.LMDBDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LatencyStatsDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LatencyStatsDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "LatencyStatsDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.LatencyStatsDataset"
   }

diff --git a/...ore/tensorflow-core-api/src/bazel/api_def/api_def_LegacyParallelInterleaveDatasetV2.pbtxt b/...ore/tensorflow-core-api/src/bazel/api_def/api_def_LegacyParallelInterleaveDatasetV2.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "LegacyParallelInterleaveDatasetV2"
+  visibility: VISIBLE
   endpoint {
-    name: "data.experimental.LegacyParallelInterleaveDataset"
+    name: "data.LegacyParallelInterleaveDataset"
   }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LoadDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LoadDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "LoadDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.LoadDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_MapAndBatchDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_MapAndBatchDataset.pbtxt
@@ -1,4 +1,7 @@
 op {
   graph_op_name: "MapAndBatchDataset"
-  visibility: SKIP
+  visibility: VISIBLE
+  endpoint {
+    name: "data.MapAndBatchDataset"
+  }
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_MapAndBatchDatasetV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_MapAndBatchDatasetV2.pbtxt
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_MatchingFilesDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_MatchingFilesDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "MatchingFilesDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.MatchingFilesDataset"
   }

diff --git a/...low-core/tensorflow-core-api/src/bazel/api_def/api_def_MaxIntraOpParallelismDataset.pbtxt b/...low-core/tensorflow-core-api/src/bazel/api_def/api_def_MaxIntraOpParallelismDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "MaxIntraOpParallelismDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.MaxIntraOpParallelismDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ModelDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ModelDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "ModelDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.ModelDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_NonSerializableDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_NonSerializableDataset.pbtxt
@@ -1,5 +1,6 @@
 op {
   graph_op_name: "NonSerializableDataset"
+  visibility: VISIBLE
   endpoint {
     name: "data.NonSerializableDataset"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_OptimizeDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_OptimizeDataset.pbtxt
@@ -1,6 +1,4 @@
 op {
   graph_op_name: "OptimizeDataset"
-  endpoint {
-    name: "data.OptimizeDataset"
-  }
+  visibility: SKIP
 }
diff --git a/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_OptimizeDatasetV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_OptimizeDatasetV2.pbtxt
@@ -1,6 +1,7 @@
 op {
   graph_op_name: "OptimizeDatasetV2"
+  visibility: VISIBLE
   endpoint {
-    name: "data.OptimizeDatasetV2"
+    name: "data.OptimizeDataset"
   }
 }