Add pretty printing.

Author: khatchad

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

@@ -17,6 +17,8 @@
  */
 public class Constant extends TensorGenerator {
 
+  private static final String FUNCTION_NAME = "tf.constant()";
+
   private static final int VALUE_NUMBER_FOR_VALUE_ARGUMENT = 2;
 
   private static final int VALUE_NUMBER_FOR_DTYPE_ARGUMENT = 3;
@@ -57,4 +59,9 @@ protected int getValueNumberForShapeArgument() {
     // function's name).
     return VALUE_NUMBER_FOR_SHAPE_ARGUMENT;
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -15,6 +15,8 @@
  */
 public class Fill extends Constant {
 
+  private static final String FUNCTION_NAME = "tf.fill()";
+
   private static final int VALUE_NUMBER_FOR_SHAPE_ARGUMENT = 2;
 
   private static final int VALUE_NUMBER_FOR_VALUE_ARGUMENT = 3;
@@ -48,4 +50,9 @@ protected int getValueNumberForShapeArgument() {
   protected Set<List<Dimension<?>>> getDefaultShapes(PropagationCallGraphBuilder builder) {
     throw new UnsupportedOperationException("Shape is mandatory and must be provided explicitly.");
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -12,7 +12,14 @@
  */
 public class Normal extends Uniform {
 
+  private static final String FUNCTION_NAME = "tf.random.normal()";
+
   public Normal(PointsToSetVariable source, CGNode node) {
     super(source, node);
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -22,6 +22,8 @@ public class Ones extends TensorGenerator {
 
   private static final java.util.logging.Logger LOGGER = getLogger(Ones.class.getName());
 
+  private static final String FUNCTION_NAME = "tf.ones()";
+
   private static final int VALUE_NUMBER_FOR_SHAPE_ARGUMENT = 2;
 
   private static final int VALUE_NUMBER_FOR_DTYPE_ARGUMENT = 3;
@@ -53,4 +55,9 @@ protected int getValueNumberForShapeArgument() {
   protected int getValueNumberForDTypeArgument() {
     return VALUE_NUMBER_FOR_DTYPE_ARGUMENT;
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -41,6 +41,8 @@ public class Range extends TensorGenerator {
 
   private static final Logger LOGGER = Logger.getLogger(Range.class.getName());
 
+  private static final String FUNCTION_NAME = "tf.range()";
+
   public Range(PointsToSetVariable source, CGNode node) {
     super(source, node);
   }
@@ -212,4 +214,9 @@ protected int getValueNumberForDTypeArgument() {
 
     return -1; // Positional dtype argument for range() is not yet implemented.
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -589,4 +589,16 @@ private EnumSet<DType> getDTypesOfValue(
   protected CGNode getNode() {
     return this.node;
   }
+
+  @Override
+  public String toString() {
+    return this.getSignature();
+  }
+
+  /**
+   * Returns the TensorFlow function signature represented by this generator.
+   *
+   * @return The TensorFlow function signature represented by this generator.
+   */
+  protected abstract String getSignature();
 }

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

@@ -13,8 +13,15 @@
  */
 public class TruncatedNormal extends Normal {
 
+  private static final String FUNCTION_NAME = "tf.random.truncated_normal()";
+
   public TruncatedNormal(PointsToSetVariable source, CGNode node) {
     super(source, node);
     // TODO Auto-generated constructor stub
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -12,6 +12,8 @@
  */
 public class Uniform extends Ones {
 
+  private static final String FUNCTION_NAME = "tf.random.uniform()";
+
   private static final int VALUE_NUMBER_FOR_DTYPE_ARGUMENT = 5;
 
   public Uniform(PointsToSetVariable source, CGNode node) {
@@ -22,4 +24,9 @@ public Uniform(PointsToSetVariable source, CGNode node) {
   protected int getValueNumberForDTypeArgument() {
     return VALUE_NUMBER_FOR_DTYPE_ARGUMENT;
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -11,7 +11,14 @@
  */
 public class Zeros extends Ones {
 
+  private static final String FUNCTION_NAME = "tf.zeros()";
+
   public Zeros(PointsToSetVariable source, CGNode node) {
     super(source, node);
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }

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

@@ -12,6 +12,8 @@
  */
 public class ZerosLike extends Constant {
 
+  public static final String FUNCTION_NAME = "tf.zeros_like()";
+
   /**
    * The shape argument is not explicitly provided to zeros_like(); rather, the shape is inferred
    * from the `input` argument.
@@ -26,4 +28,9 @@ public ZerosLike(PointsToSetVariable source, CGNode node) {
   protected int getValueNumberForShapeArgument() {
     return VALUE_NUMBER_FOR_SHAPE_ARGUMENT;
   }
+
+  @Override
+  protected String getSignature() {
+    return FUNCTION_NAME;
+  }
 }