assume only graph or eager environments

Signed-off-by: Ryan Nett <[email protected]>

Author: rnett

tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/op/core/Init.java

@@ -98,14 +98,10 @@ public static Init create(Scope scope) {
   @Endpoint(name = "initAdd")
   public static void add(Scope scope, Op initializer) {
     ExecutionEnvironment exEnv = scope.env();
-    if(exEnv.isEager()){
-      return;
-    }
-    if (!(exEnv instanceof Graph)) {
-      throw new IllegalArgumentException("initAdd is only supported on Graph or eager sessions.");
+
+    if(exEnv.isGraph()) {
+      ((Graph) exEnv).addInitializer(initializer);
     }
-    Graph graph = (Graph) exEnv;
-    graph.addInitializer(initializer);
   }
 
   private Init(Operation operation) {