Better dependency resolution

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

Author: rnett

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

@@ -76,10 +76,21 @@ public synchronized List<String> getDependencies() {
       Set<String> deps = new LinkedHashSet<>();
 
       for (NodeDef node : getFunctionDef().getNodeDefList()) {
-        if (node.getOp().equals(ConcreteFunction.CALL_OP)
-            || node.getOp().equals(ConcreteFunction.STATEFUL_CALL_OP)) {
-          deps.add(node.getAttrMap().get("f").getFunc().getName());
-        }
+        node.getAttrMap()
+            .values()
+            .forEach(
+                (attr) -> {
+                  if (attr.hasFunc()) {
+                    deps.add(attr.getFunc().getName());
+                  } else if (attr.hasList()) {
+                    attr.getList()
+                        .getFuncList()
+                        .forEach(
+                            funcs -> {
+                              deps.add(funcs.getName());
+                            });
+                  }
+                });
       }
       dependencies = Collections.unmodifiableList(new ArrayList<>(deps));
     }