Add API tf.add().

Author: khatchad

com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java

@@ -107,8 +107,8 @@ public void testTf2()
     testTf2("tf2n.py", "func2", 1, 1, 2);
     testTf2("tf2n2.py", "func2", 1, 1, 2);
     testTf2("tf2n3.py", "func2", 1, 1, 2);
-    testTf2("tf2o.py", "add", 2, 2, 2, 3);
-    testTf2("tf2o2.py", "add", 2, 2, 2, 3);
+    testTf2("tf2o.py", "add", 2, 3, 2, 3);
+    testTf2("tf2o2.py", "add", 2, 3, 2, 3);
     testTf2("tf2p.py", "value_index", 2, 2, 2, 3);
     testTf2("tf2p2.py", "value_index", 2, 2, 2, 3);
     testTf2("tf2q.py", "add", 2, 2, 2, 3);
@@ -245,12 +245,18 @@ public void testTf2()
         3); // NOTE: Change to 2 tensor parameters and 5 tensor variables once
     // https://github.com/wala/ML/issues/127 is fixed. Values 2 and 3 will correspond to the
     // tensor parameters.
-    testTf2("autoencoder.py", "encoder", 1, 10, 2);
-    testTf2("autoencoder.py", "mean_square", 1, 1, 3);
-    testTf2("autoencoder.py", "run_optimization", 1, 1, 2);
-    testTf2("autoencoder.py", "decoder", 1, 8, 2);
+    testTf2("autoencoder.py", "encoder", 1, 18, 2);
+    testTf2("autoencoder.py", "mean_square", 2, 2, 2, 3);
+    testTf2("autoencoder.py", "run_optimization", 1, 3, 2);
+    testTf2("autoencoder.py", "decoder", 1, 18, 2);
     testTf2("tf2_test_sigmoid.py", "f", 1, 1, 2);
     testTf2("tf2_test_sigmoid2.py", "f", 1, 1, 2);
+    testTf2("tf2_test_add.py", "f", 1, 1, 2);
+    testTf2("tf2_test_add2.py", "f", 1, 1, 2);
+    testTf2("tf2_test_add3.py", "f", 1, 1, 2);
+    testTf2("tf2_test_add4.py", "f", 1, 1, 2);
+    testTf2("tf2_test_add5.py", "f", 1, 1, 2);
+    testTf2("tf2_test_add6.py", "f", 1, 1, 2);
   }
 
   private void testTf2(

com.ibm.wala.cast.python.ml/data/tensorflow.xml

@@ -74,6 +74,9 @@
         <new def="sigmoid" class="Ltensorflow/math/sigmoid" />
         <putfield class="LRoot" field="sigmoid" fieldType="LRoot" ref="nn" value="sigmoid" />
         <putfield class="LRoot" field="sigmoid" fieldType="LRoot" ref="math" value="sigmoid" />
+        <new def="add" class="Ltensorflow/math/add" />
+        <putfield class="LRoot" field="add" fieldType="LRoot" ref="x" value="add" />
+        <putfield class="LRoot" field="add" fieldType="LRoot" ref="math" value="add" />
         <new def="placeholder" class="Ltensorflow/functions/placeholder" />
         <putfield class="LRoot" field="placeholder" fieldType="LRoot" ref="x" value="placeholder" />
         <new def="examples" class="Lobject" />
@@ -263,6 +266,18 @@
           <return value="x" />
         </method>
       </class>
+      <class name="add" allocatable="true">
+        <!-- https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/add -->
+        <method name="read_data" descriptor="()LRoot;">
+          <new def="x" class="Ltensorflow/math/add" />
+          <return value="x" />
+        </method>
+        <method name="do" descriptor="()LRoot;" numArgs="4" paramNames="self x y name">
+          <!-- Even though tf.add() isn't a tensor "generator," it can convert its non-tensor arguments to tensors. -->
+          <call class="LRoot" name="read_data" descriptor="()LRoot;" type="virtual" arg0="arg0" def="xx" />
+          <return value="xx" />
+        </method>
+      </class>
     </package>
     <package name="tensorflow/functions">
       <class name="AdamOptimizer" allocatable="true">

com.ibm.wala.cast.python.test/data/tf2_test_add.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+x = [1, 2, 3, 4, 5]
+y = 1
+z = tf.add(x, y)
+f(z)

com.ibm.wala.cast.python.test/data/tf2_test_add2.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+x = tf.convert_to_tensor([1, 2, 3, 4, 5])
+y = tf.convert_to_tensor(1)
+z = tf.add(x, y)
+f(z)

com.ibm.wala.cast.python.test/data/tf2_test_add3.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+x = [1, 2, 3, 4, 5]
+y = tf.constant([1, 2, 3, 4, 5])
+z = tf.add(x, y)
+f(z)

com.ibm.wala.cast.python.test/data/tf2_test_add4.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+x = [1, 2, 3, 4, 5]
+y = 1
+z = tf.math.add(x, y)
+f(z)

com.ibm.wala.cast.python.test/data/tf2_test_add5.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+x = tf.convert_to_tensor([1, 2, 3, 4, 5])
+y = tf.convert_to_tensor(1)
+z = tf.math.add(x, y)
+f(z)

com.ibm.wala.cast.python.test/data/tf2_test_add6.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+x = [1, 2, 3, 4, 5]
+y = tf.constant([1, 2, 3, 4, 5])
+z = tf.math.add(x, y)
+f(z)