Workaround wala/ML#195.

ponder-lab · May 9, 2024 · b1d8015 · b1d8015
1 parent 2520249
commit b1d8015
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/edu.cuny.hunter.hybridize.core/tensorflow.xml b/edu.cuny.hunter.hybridize.core/tensorflow.xml
@@ -512,12 +512,11 @@
         </method>
       </class>
       <class name="reshape" allocatable="true">
-        <method name="copy_data" descriptor="()LRoot;">
-          <new def="x" class="Ltensorflow/examples/tutorials/mnist/dataset" />
-          <return value="x" />
-        </method>
-        <method name="do" descriptor="()LRoot;" numArgs="3">
-          <call class="LRoot" name="copy_data" descriptor="()LRoot;" type="virtual" arg0="arg0" def="x" />
+        <!-- https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/reshape -->
+        <method name="do" descriptor="()LRoot;" numArgs="2" paramNames="input name">
+          <!-- Workaround https://github.com/wala/ML/issues/195 -->
+          <new def="z" class="Ltensorflow/functions/convert_to_tensor" />
+          <call class="Ltensorflow/functions/convert_to_tensor" name="do" descriptor="()LRoot;" type="virtual" arg0="z" arg1="input" def="x" />
           <return value="x" />
         </method>
       </class>