Progress.

Author: khatchad

com.ibm.wala.cast.python.ml/source/com/ibm/wala/cast/python/ml/analysis/TensorTypeAnalysis.java

@@ -447,11 +447,11 @@ public String toString() {
     };
   }
 
-  private final Map<PointsToSetVariable, TensorType> init;
+  private final Map<PointsToSetVariable, Set<TensorType>> init;
 
   public TensorTypeAnalysis(
       Graph<PointsToSetVariable> G,
-      Map<PointsToSetVariable, TensorType> init,
+      Map<PointsToSetVariable, Set<TensorType>> init,
       Map<PointsToSetVariable, TensorType> reshapeTypes,
       Map<PointsToSetVariable, TensorType> set_shapes,
       Set<PointsToSetVariable> conv2ds,
@@ -480,7 +480,8 @@ protected TensorVariable[] makeStmtRHS(int size) {
   protected void initializeVariables() {
     super.initializeVariables();
     for (PointsToSetVariable src : init.keySet()) {
-      getOut(src).state.add(init.get(src));
+      Set<TensorType> tensorTypes = init.get(src);
+      getOut(src).state.addAll(tensorTypes);
     }
   }
 

com.ibm.wala.cast.python.ml/source/com/ibm/wala/cast/python/ml/client/PythonTensorAnalysisEngine.java

@@ -715,9 +715,8 @@ public TensorTypeAnalysis performAnalysis(PropagationCallGraphBuilder builder)
 
     Set<PointsToSetVariable> conv3ds = getKeysDefinedByCall(conv3d, builder);
 
-    TensorTypeAnalysis tt = null;
-    //        new TensorTypeAnalysis(dataflow, init, shapeOps, setCalls, conv2ds, conv3ds,
-    // errorLog);
+    TensorTypeAnalysis tt =
+        new TensorTypeAnalysis(dataflow, init, shapeOps, setCalls, conv2ds, conv3ds, errorLog);
 
     tt.solve(new NullProgressMonitor());