Adjust GraphOperation#input to not require a graph lock

Signed-off-by: Ryan Nett <JNett96@gmail.com>

Author: rnett

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/GraphOperation.java

@@ -213,8 +213,7 @@ public Output<?> input(int idx) {
     try (PointerScope scope = new PointerScope()) {
       TF_Input input = new TF_Input().oper(getUnsafeNativeHandle()).index(idx);
       TF_Output output = TF_OperationInput(input);
-      String opName = TF_OperationName(output.oper()).getString();
-      return graph.operation(opName).output(output.index());
+      return new GraphOperation(graph, output.oper()).output(output.index());
     }
   }
 

tensorflow-core/tensorflow-core-api/src/test/java/org/tensorflow/CustomGradientTest.java

@@ -22,10 +22,6 @@
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import java.util.Arrays;
-import java.util.Spliterator;
-import java.util.Spliterators;
-import java.util.stream.Collectors;
-import java.util.stream.StreamSupport;
 import org.junit.jupiter.api.Test;
 import org.tensorflow.ndarray.index.Indices;
 import org.tensorflow.op.Ops;
@@ -68,13 +64,13 @@ public void testCustomGradient() {
       assertEquals(1, grads0.length);
       assertEquals(DataType.DT_FLOAT, grads0[0].dataType());
 
-      System.out.println(
-          StreamSupport.stream(
-                  Spliterators.spliteratorUnknownSize(
-                      g.operations(), Spliterator.ORDERED | Spliterator.NONNULL),
-                  false)
-              .map(GraphOperation::name)
-              .collect(Collectors.toList()));
+      //      System.out.println(
+      //          StreamSupport.stream(
+      //                  Spliterators.spliteratorUnknownSize(
+      //                      g.operations(), Spliterator.ORDERED | Spliterator.NONNULL),
+      //                  false)
+      //              .map(GraphOperation::name)
+      //              .collect(Collectors.toList()));
 
       try (TFloat32 c1 = TFloat32.vectorOf(3.0f, 2.0f, 1.0f, 0.0f);
           AutoCloseableList<Tensor> outputs =