We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 517fd8d commit f5dd0e5Copy full SHA for f5dd0e5
tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/CustomGradient.java
@@ -30,6 +30,9 @@
30
* reflection. To operate on the {@link org.tensorflow.GraphOperation} directly use {@link
31
* RawCustomGradient}.
32
*
33
+ * <p>The type of the op is not checked here, but it is required to match the class given to the
34
+ * adapter.
35
+ *
36
* @param <T> the type of op this gradient is for.
37
*/
38
@SuppressWarnings("rawtypes")
0 commit comments