Document use of rawtypes

Signed-off-by: Ryan Nett <[email protected]>
tensorflow · Nov 11, 2021 · f5dd0e5 · f5dd0e5
1 parent 517fd8d
commit f5dd0e5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/CustomGradient.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/CustomGradient.java
@@ -30,6 +30,9 @@
  * reflection. To operate on the {@link org.tensorflow.GraphOperation} directly use {@link
  * RawCustomGradient}.
  *
+ * <p>The type of the op is not checked here, but it is required to match the class given to the
+ * adapter.
+ *
  * @param <T> the type of op this gradient is for.
  */
 @SuppressWarnings("rawtypes")