diff options
Diffstat (limited to 'tensorflow/java/src/main/java/org/tensorflow/op/core/Gradients.java')
-rw-r--r-- | tensorflow/java/src/main/java/org/tensorflow/op/core/Gradients.java | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/java/src/main/java/org/tensorflow/op/core/Gradients.java b/tensorflow/java/src/main/java/org/tensorflow/op/core/Gradients.java index 6d71ddfff0..5432ff244e 100644 --- a/tensorflow/java/src/main/java/org/tensorflow/op/core/Gradients.java +++ b/tensorflow/java/src/main/java/org/tensorflow/op/core/Gradients.java @@ -88,7 +88,11 @@ public class Gradients implements Op, Iterable<Operand<?>> { } } } - Output<?>[] dy = scope.graph().addGradients(scope.prefix(), Operands.asOutputs(y), Operands.asOutputs(x), dx); + Output<?>[] dy = scope.graph().addGradients( + scope.makeOpName("Gradients"), + Operands.asOutputs(y), + Operands.asOutputs(x), + dx); return new Gradients(Arrays.asList(dy)); } |