From f5dd0e573d0d02b030a43da3149c96e7a5994c45 Mon Sep 17 00:00:00 2001
From: Ryan Nett <JNett96@gmail.com>
Date: Wed, 10 Nov 2021 16:52:48 -0800
Subject: [PATCH] Document use of rawtypes

Signed-off-by: Ryan Nett <JNett96@gmail.com>
---
 .../src/main/java/org/tensorflow/op/CustomGradient.java        | 3 +++
 1 file changed, 3 insertions(+)

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
index 345787265da..2d573808222 100644
--- 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")