From e50f7111d3ddf7602a0cf08d8d9cc281ec376aa3 Mon Sep 17 00:00:00 2001
From: Raffi Khatchadourian <khatchad@hunter.cuny.edu>
Date: Tue, 23 Jan 2024 11:55:19 -0500
Subject: [PATCH] New API.

---
 com.ibm.wala.cast.python.ml/data/tensorflow.xml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/com.ibm.wala.cast.python.ml/data/tensorflow.xml b/com.ibm.wala.cast.python.ml/data/tensorflow.xml
index e4f30ff9b..01b98e231 100644
--- a/com.ibm.wala.cast.python.ml/data/tensorflow.xml
+++ b/com.ibm.wala.cast.python.ml/data/tensorflow.xml
@@ -93,6 +93,9 @@
         <new def="softmax" class="Ltensorflow/functions/softmax" />
         <putfield class="LRoot" field="softmax" fieldType="LRoot" ref="nn" value="softmax" />
 
+        <new def="sigmoid" class="Ltensorflow/functions/sigmoid" />
+        <putfield class="LRoot" field="sigmoid" fieldType="LRoot" ref="nn" value="sigmoid" />
+
         <new def="placeholder" class="Ltensorflow/functions/placeholder" />
         <putfield class="LRoot" field="placeholder" fieldType="LRoot" ref="x" value="placeholder" />
 
@@ -781,6 +784,13 @@
           <return value="logits" />
         </method>
       </class>
+
+      <class name="sigmoid" allocatable="true">
+        <!-- https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/nn/sigmoid -->
+        <method name="do" descriptor="()LRoot;" numArgs="3" paramNames="self x name">
+          <return value="x" />
+        </method>
+      </class>
     </package>
 
     <package name="tensorflow/estimator">