Allow placeholders in Graph's init scope (#393)

rnett · web-flow · commit ef72244ac47c · 2021-10-19T21:44:44.000-04:00
Signed-off-by: Ryan Nett &lt;JNett96@gmail.com&gt;
diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Graph.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Graph.java
@@ -588,10 +588,6 @@ private boolean registerInitOpHelper(Operation op) {
     }
     GraphOperation graphOp = (GraphOperation) op;
 
-    if (op.type().equals(Placeholder.OP_NAME)) {
-      throw new IllegalStateException("Can not make a placeholder " + op + " an init op.");
-    }
-
     for (GraphOperation controlInput : graphOp.controlInputs()) {
       registerInitOpHelper(controlInput);
     }

Original file line number	Diff line number	Diff line change
`@@ -588,10 +588,6 @@ private boolean registerInitOpHelper(Operation op) {`
`588`	`588`	`}`
`589`	`589`	`GraphOperation graphOp = (GraphOperation) op;`
`590`	`590`
`591`		`- if (op.type().equals(Placeholder.OP_NAME)) {`
`592`		`- throw new IllegalStateException("Can not make a placeholder " + op + " an init op.");`
`593`		`- }`
`594`		`-`
`595`	`591`	`for (GraphOperation controlInput : graphOp.controlInputs()) {`
`596`	`592`	`registerInitOpHelper(controlInput);`
`597`	`593`	`}`