Op classifications for TF 2.18.

tensorflow · karllessard · May 23, 2025 · Oct 18, 2024 · Mar 21, 2025 · Apr 11, 2025
commit b533b21c820fc0c0bfa9252c305460cec8fb2b31
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CheckPinned.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CheckPinned.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "CheckPinned"
+  endpoint {
+    name: "CheckPinned"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataSize.pbtxt
@@ -1,6 +1,6 @@
 op {
   graph_op_name: "ComputeDedupDataSize"
-  visibility: VISIBLE
+  visibility: SKIP
   endpoint {
     name: "tpu.ComputeDedupDataSize"
   }

diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataSizeV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataSizeV2.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "ComputeDedupDataSizeV2"
+  endpoint {
+    name: "tpu.ComputeDedupDataSize"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMask.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMask.pbtxt
@@ -1,5 +1,5 @@
 op {
-  visibility: VISIBLE
+  visibility: SKIP
   graph_op_name: "ComputeDedupDataTupleMask"
   endpoint {
     name: "tpu.ComputeDedupDataTupleMask"

diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMaskV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMaskV2.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "ComputeDedupDataTupleMaskV2"
+  endpoint {
+    name: "tpu.ComputeDedupDataTupleMask"
+  }
+}
diff --git a/...orflow-core/tensorflow-core-api/src/api/api_def_ConvertToListOfSparseCoreCooTensors.pbtxt b/...orflow-core/tensorflow-core-api/src/api/api_def_ConvertToListOfSparseCoreCooTensors.pbtxt
@@ -0,0 +1,7 @@
+op {
+  graph_op_name: "ConvertToListOfSparseCoreCooTensors"
+  visibility: VISIBLE
+  endpoint {
+    name: "sparse.ConvertToListOfSparseCoreCooTensors"
+  }
+}
diff --git a/...low-core/tensorflow-core-api/src/api/api_def_ConvertToSparseCoreCsrWrappedCooTensor.pbtxt b/...low-core/tensorflow-core-api/src/api/api_def_ConvertToSparseCoreCsrWrappedCooTensor.pbtxt
@@ -0,0 +1,7 @@
+op {
+  graph_op_name: "ConvertToSparseCoreCsrWrappedCooTensor"
+  visibility: VISIBLE
+  endpoint {
+    name: "sparse.ConvertToSparseCoreCsrWrappedCooTensor"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbedding.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbedding.pbtxt
@@ -1,5 +1,5 @@
 op {
-  visibility: VISIBLE
+  visibility: SKIP
   graph_op_name: "FinalizeTPUEmbedding"
   endpoint {
     name: "tpu.FinalizeTPUEmbedding"

diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbeddingV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbeddingV2.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "FinalizeTPUEmbeddingV2"
+  endpoint {
+    name: "tpu.FinalizeTPUEmbedding"
+  }
+}
diff --git a/...low-core/tensorflow-core-api/src/api/api_def_GetStatsFromListOfSparseCoreCooTensors.pbtxt b/...low-core/tensorflow-core-api/src/api/api_def_GetStatsFromListOfSparseCoreCooTensors.pbtxt
@@ -0,0 +1,7 @@
+op {
+  graph_op_name: "GetStatsFromListOfSparseCoreCooTensors"
+  visibility: VISIBLE
+  endpoint {
+    name: "sparse.GetStatsFromListOfSparseCoreCooTensors"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetTpuTaskId.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetTpuTaskId.pbtxt
@@ -0,0 +1,7 @@
+op {
+  graph_op_name: "GetTpuTaskId"
+  visibility: VISIBLE
+  endpoint {
+    name: "tpu.GetTpuTaskId"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GlobalShuffleDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GlobalShuffleDataset.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "GlobalShuffleDataset"
+  endpoint {
+    name: "data.GlobalShuffleDataset"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IndexFlatMapDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IndexFlatMapDataset.pbtxt
@@ -0,0 +1,7 @@
+op {
+  graph_op_name: "IndexFlatMapDataset"
+  visibility: VISIBLE
+  endpoint {
+    name: "data.IndexFlatMapDataset"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetModelProto.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetModelProto.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "IteratorGetModelProto"
+  endpoint {
+    name: "data.IteratorGetModelProto"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SortListOfSparseCoreCooTensors.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SortListOfSparseCoreCooTensors.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "SortListOfSparseCoreCooTensors"
+  endpoint {
+    name: "sparse.SortListOfSparseCoreCooTensors"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UpdateTaskIdAndGlobalCoreArray.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UpdateTaskIdAndGlobalCoreArray.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "UpdateTaskIdAndGlobalCoreArray"
+  endpoint {
+    name: "tpu.UpdateTaskIdAndGlobalCoreArray"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WeightedFlatMapDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WeightedFlatMapDataset.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "WeightedFlatMapDataset"
+  endpoint {
+    name: "data.WeightedFlatMapDataset"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivations.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivations.pbtxt
@@ -1,5 +1,5 @@
 op {
-  visibility: VISIBLE
+  visibility: SKIP
   graph_op_name: "XlaRecvTPUEmbeddingActivations"
   endpoint {
     name: "xla.XlaRecvTPUEmbeddingActivations"

diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivationsV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivationsV2.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaRecvTPUEmbeddingActivationsV2"
+  endpoint {
+    name: "xla.XlaRecvTPUEmbeddingActivations"
+  }
+}
diff --git a/...rflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt b/...rflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt
@@ -1,5 +1,5 @@
 op {
-  visibility: VISIBLE
+  visibility: SKIP
   graph_op_name: "XlaRecvTPUEmbeddingDeduplicationData"
   endpoint {
     name: "xla.XlaRecvTPUEmbeddingDeduplicationData"

diff --git a/...low-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationDataV2.pbtxt b/...low-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationDataV2.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaRecvTPUEmbeddingDeduplicationDataV2"
+  endpoint {
+    name: "xla.XlaRecvTPUEmbeddingDeduplicationData"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradients.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradients.pbtxt
@@ -1,5 +1,5 @@
 op {
-  visibility: VISIBLE
+  visibility: SKIP
   graph_op_name: "XlaSendTPUEmbeddingGradients"
   endpoint {
     name: "xla.XlaSendTPUEmbeddingGradients"

diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradientsV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradientsV2.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSendTPUEmbeddingGradientsV2"
+  endpoint {
+    name: "xla.XlaSendTPUEmbeddingGradients"
+  }
+}
diff --git a/...low-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithAdagradAndStaticBufferSize.pbtxt b/...low-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithAdagradAndStaticBufferSize.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulGradWithAdagradAndStaticBufferSize"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulGradWithAdagradAndStaticBufferSize"
+  }
+}
diff --git a/...-api/src/api/api_def_XlaSparseDenseMatmulGradWithAdagradMomentumAndStaticBufferSize.pbtxt b/...-api/src/api/api_def_XlaSparseDenseMatmulGradWithAdagradMomentumAndStaticBufferSize.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulGradWithAdagradMomentumAndStaticBufferSize"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulGradWithAdagradMomentumAndStaticBufferSize"
+  }
+}
diff --git a/...orflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithAdamAndStaticBufferSize.pbtxt b/...orflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithAdamAndStaticBufferSize.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulGradWithAdamAndStaticBufferSize"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulGradWithAdamAndStaticBufferSize"
+  }
+}
diff --git a/...rflow-core/tensorflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithCsrInput.pbtxt b/...rflow-core/tensorflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithCsrInput.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulGradWithCsrInput"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulGradWithCsrInput"
+  }
+}
diff --git a/...orflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithFtrlAndStaticBufferSize.pbtxt b/...orflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithFtrlAndStaticBufferSize.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulGradWithFtrlAndStaticBufferSize"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulGradWithFtrlAndStaticBufferSize"
+  }
+}
diff --git a/...sorflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithSgdAndStaticBufferSize.pbtxt b/...sorflow-core-api/src/api/api_def_XlaSparseDenseMatmulGradWithSgdAndStaticBufferSize.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulGradWithSgdAndStaticBufferSize"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulGradWithSgdAndStaticBufferSize"
+  }
+}
diff --git a/...w-core/tensorflow-core-api/src/api/api_def_XlaSparseDenseMatmulWithStaticBufferSize.pbtxt b/...w-core/tensorflow-core-api/src/api/api_def_XlaSparseDenseMatmulWithStaticBufferSize.pbtxt
@@ -0,0 +1,6 @@
+op {
+  graph_op_name: "XlaSparseDenseMatmulWithStaticBufferSize"
+  endpoint {
+    name: "xla.XlaSparseDenseMatmulWithStaticBufferSize"
+  }
+}
diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/BitwiseOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/BitwiseOps.java
@@ -61,7 +61,6 @@ public final class BitwiseOps {
    *    tf.assert_equal(tf.cast(res, tf.float32), exp) # TRUE
    *  </pre>
    *
-   * @param <T> data type for {@code z} output
    * @param x The x value
    * @param y The y value
    * @param <T> data type for {@code BitwiseAnd} output and operands
@@ -91,7 +90,6 @@ public <T extends TNumber> BitwiseAnd<T> bitwiseAnd(Operand<T> x, Operand<T> y)
    *    tf.assert_equal(tf.cast(res,  tf.float32), exp)  # TRUE
    *  </pre>
    *
-   * @param <T> data type for {@code z} output
    * @param x The x value
    * @param y The y value
    * @param <T> data type for {@code BitwiseOr} output and operands
@@ -121,7 +119,6 @@ public <T extends TNumber> BitwiseOr<T> bitwiseOr(Operand<T> x, Operand<T> y) {
    *    tf.assert_equal(tf.cast(res, tf.float32), exp) # TRUE
    *  </pre>
    *
-   * @param <T> data type for {@code z} output
    * @param x The x value
    * @param y The y value
    * @param <T> data type for {@code BitwiseXor} output and operands
@@ -172,7 +169,6 @@ public <T extends TNumber> BitwiseXor<T> bitwiseXor(Operand<T> x, Operand<T> y)
    *      tf.assert_equal(tf.cast(inverted, tf.float32), tf.cast(expected, tf.float32))
    *  </pre>
    *
-   * @param <T> data type for {@code y} output
    * @param x The x value
    * @param <T> data type for {@code Invert} output and operands
    * @return a new instance of Invert
@@ -212,7 +208,6 @@ public <T extends TNumber> Invert<T> invert(Operand<T> x) {
    *  # &lt;tf.Tensor: shape=(4,), dtype=int8, numpy=array([ -2,  64, 101,  32], dtype=int8)&gt;
    *  </pre>
    *
-   * @param <T> data type for {@code z} output
    * @param x The x value
    * @param y The y value
    * @param <T> data type for {@code LeftShift} output and operands
@@ -255,7 +250,6 @@ public <T extends TNumber> LeftShift<T> leftShift(Operand<T> x, Operand<T> y) {
    *  # &lt;tf.Tensor: shape=(4,), dtype=int8, numpy=array([ -2,  64, 101,  32], dtype=int8)&gt;
    *  </pre>
    *
-   * @param <T> data type for {@code z} output
    * @param x The x value
    * @param y The y value
    * @param <T> data type for {@code RightShift} output and operands

diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/CollectiveOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/CollectiveOps.java
@@ -49,7 +49,6 @@ public final class CollectiveOps {
   /**
    * Mutually exchanges multiple tensors of identical type and shape.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param communicator The communicator value
    * @param groupAssignment The groupAssignment value
@@ -79,7 +78,6 @@ public CollectiveAssignGroup collectiveAssignGroup(Operand<TInt32> groupAssignme
   /**
    * Receives a tensor value broadcast from another device.
    *
-   * @param <U> data type for {@code data} output
    * @param groupSize The groupSize value
    * @param groupKey The groupKey value
    * @param instanceKey The instanceKey value
@@ -98,7 +96,6 @@ public <U extends TType> CollectiveBcastRecv<U> collectiveBcastRecv(Operand<TInt
   /**
    * Broadcasts a tensor value to one or more other devices.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param groupSize The groupSize value
    * @param groupKey The groupKey value
@@ -119,7 +116,6 @@ public <T extends TType> CollectiveBcastSend<T> collectiveBcastSend(Operand<T> i
    *  collective ops. In this case, keys that are unique at runtime
    *  (e.g. {@code instance_key}) should be used to distinguish collective groups.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param groupSize The groupSize value
    * @param groupKey The groupKey value
@@ -157,7 +153,6 @@ public CollectiveInitializeCommunicator collectiveInitializeCommunicator(Operand
    *  source_target_pairs={@code [[0,1],[1,2],[2,3],[3,0]]} gets the outputs:
    *  {@code [D, A, B, C]}.
    *
-   * @param <T> data type for {@code output} output
    * @param input The local input to be permuted. Currently only supports float and
    *  bfloat16.
    * @param sourceTargetPairs A tensor with shape [num_pairs, 2].
@@ -172,7 +167,6 @@ public <T extends TType> CollectivePermute<T> collectivePermute(Operand<T> input
   /**
    * Mutually reduces multiple tensors of identical type and shape.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param communicator The communicator value
    * @param groupAssignment The groupAssignment value
@@ -193,7 +187,6 @@ public <T extends TNumber> CollectiveReduce<T> collectiveReduce(Operand<T> input
    *  collective ops. In this case, keys that are unique at runtime
    *  (e.g. {@code instance_key}) should be used to distinguish collective groups.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param groupSize The groupSize value
    * @param groupKey The groupKey value

diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataOps.java
@@ -59,6 +59,7 @@
 import org.tensorflow.op.data.GroupByReducerDataset;
 import org.tensorflow.op.data.GroupByWindowDataset;
 import org.tensorflow.op.data.IgnoreErrorsDataset;
+import org.tensorflow.op.data.IndexFlatMapDataset;
 import org.tensorflow.op.data.InitializeTableFromDataset;
 import org.tensorflow.op.data.InterleaveDataset;
 import org.tensorflow.op.data.Iterator;
@@ -819,6 +820,28 @@ public IgnoreErrorsDataset ignoreErrorsDataset(Operand<? extends TType> inputDat
     return IgnoreErrorsDataset.create(scope, inputDataset, outputTypes, outputShapes, options);
   }
 
+  /**
+   * The IndexFlatMapDataset operation
+   *
+   * @param inputDataset The inputDataset value
+   * @param mapFuncOtherArgs The mapFuncOtherArgs value
+   * @param indexMapFuncOtherArgs The indexMapFuncOtherArgs value
+   * @param outputCardinality The outputCardinality value
+   * @param mapFunc The value of the mapFunc attribute
+   * @param indexMapFunc The value of the indexMapFunc attribute
+   * @param outputTypes The value of the outputTypes attribute
+   * @param outputShapes The value of the outputShapes attribute
+   * @param options carries optional attribute values
+   * @return a new instance of IndexFlatMapDataset
+   */
+  public IndexFlatMapDataset indexFlatMapDataset(Operand<? extends TType> inputDataset,
+      Iterable<Operand<?>> mapFuncOtherArgs, Iterable<Operand<?>> indexMapFuncOtherArgs,
+      Operand<TInt64> outputCardinality, ConcreteFunction mapFunc, ConcreteFunction indexMapFunc,
+      List<Class<? extends TType>> outputTypes, List<Shape> outputShapes,
+      IndexFlatMapDataset.Options... options) {
+    return IndexFlatMapDataset.create(scope, inputDataset, mapFuncOtherArgs, indexMapFuncOtherArgs, outputCardinality, mapFunc, indexMapFunc, outputTypes, outputShapes, options);
+  }
+
   /**
    * The InitializeTableFromDataset operation
    *
@@ -987,7 +1010,6 @@ public LatencyStatsDataset latencyStatsDataset(Operand<? extends TType> inputDat
   /**
    * Computes rectified linear gradients for a LeakyRelu operation.
    *
-   * @param <T> data type for {@code backprops} output
    * @param gradients The backpropagated gradients to the corresponding LeakyRelu operation.
    * @param features The features passed as input to the corresponding LeakyRelu operation,
    *  OR the outputs of that operation (both work equivalently).

diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DebuggingOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DebuggingOps.java
@@ -43,7 +43,6 @@ public final class DebuggingOps {
    *  tensor. Unlike CheckNumerics (V1), CheckNumericsV2 distinguishes -Inf and +Inf
    *  in the errors it throws.
    *
-   * @param <T> data type for {@code output} output
    * @param tensor The tensor value
    * @param message Prefix of the error message.
    * @param <T> data type for {@code CheckNumericsV2} output and operands

diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DistributeOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DistributeOps.java
@@ -52,7 +52,6 @@ public final class DistributeOps {
    *  num_devices: The number of devices participating in this reduction.
    *  shared_name: Identifier that shared between ops of the same reduction.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param reduction The value of the reduction attribute
    * @param numDevices The value of the numDevices attribute
@@ -74,7 +73,6 @@ public <T extends TNumber> NcclAllReduce<T> ncclAllReduce(Operand<T> input, Stri
    *  output: The same as input.
    *  shape: The shape of the input tensor.
    *
-   * @param <T> data type for {@code output} output
    * @param input The input value
    * @param shape The value of the shape attribute
    * @param <T> data type for {@code NcclBroadcast} output and operands
@@ -93,7 +91,6 @@ public <T extends TNumber> NcclBroadcast<T> ncclBroadcast(Operand<T> input, Shap
    *  data: the value of the reduction across all {@code num_devices} devices.
    *  reduction: the reduction operation to perform.
    *
-   * @param <T> data type for {@code data} output
    * @param input The input value
    * @param reduction The value of the reduction attribute
    * @param <T> data type for {@code NcclReduce} output and operands