Extract tf.function decorator keyword arguments

edu.cuny.hunter.hybridize.core/src/edu/cuny/hunter/hybridize/core/analysis/TensorSpec.java

@@ -0,0 +1,90 @@
+package edu.cuny.hunter.hybridize.core.analysis;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
+
+/**
+ * A representation of a tf.Tensorspec which describes a tf.Tensor
+ */
+public class TensorSpec {
+
+	public enum Dtype {
+		float16, float32, float64, int32, int64, uint8, uint16, uint32, uint64, int16, int8, complex64, complex128, string, bool, qint8,
+		quint8, qint16, quint16, qint32, bfloat16, half, resource, variant
+	}
+
+	/**
+	 * Shape of the tensor being described by {@link TensorSpec}.
+	 */
+	private List<Integer> shape;
+
+	/**
+	 * Type of the tensor being described by {@link TensorSpec}.
+	 */
+	private Dtype dtype;
+
+	public TensorSpec() {
+		// Initialize to empty list
+		this.shape = new ArrayList<>();
+		this.dtype = Dtype.float32;
+	}
+
+	public TensorSpec(List<Integer> s, Dtype d) {
+		this.shape = s;
+		this.dtype = d;
+	}
+
+	/**
+	 * Shape of {@link TensorSpec}.
+	 *
+	 * @return List of dimensions (null if the shape is unspecified) of this {@link TensorSpec} shape.
+	 */
+	public List<Integer> getShape() {
+		return this.shape;
+	}
+
+	/**
+	 * Dtype of {@link TensorSpec}.
+	 *
+	 * @return Dtype of this {@link TensorSpec} dtype.
+	 */
+	public Dtype getDType() {
+		return this.dtype;
+	}
+
+	/**
+	 * Set shape of {@link TensorSpec}.
+	 */
+	public void setShape(List<Integer> s) {
+		this.shape = s;
+	}
+
+	/**
+	 * Set dtype of {@link TensorSpec}.
+	 */
+	public void setDType(Dtype d) {
+		this.dtype = d;
+	}
+
+	@Override
+	public int hashCode() {
+		return Objects.hash(shape, dtype);
+	}
+
+	@Override
+	public boolean equals(Object tensorObject) {
+
+		if (tensorObject == this) {
+			return true;
+		}
+
+		if (!(tensorObject instanceof TensorSpec)) {
+			return false;
+		}
+
+		TensorSpec tensor = (TensorSpec) tensorObject;
+
+		return shape.equals(tensor.shape) && dtype.equals(tensor.dtype);
+	}
+}

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments/in/A.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+@tf.function(input_signature=None)
+def func(x):
+  return x
+
+
+if __name__ == '__main__':
+    number = tf.constant([1.0, 1.0])
+    func(number)

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments10/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(jit_compile=None)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments10/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments11/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(reduce_retracing=True)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments11/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments12/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(reduce_retracing=False)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments12/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments13/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_implements="google.matmul_low_rank_matrix")
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments13/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments14/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_implements="embedded_matmul")
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments14/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments15/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_implements=None)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments15/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments16/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_autograph_options=tf.autograph.experimental.Feature.EQUALITY_OPERATORS)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments16/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments17/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_autograph_options=tf.autograph.experimental.Feature.ALL)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments17/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments18/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_autograph_options=(tf.autograph.experimental.Feature.EQUALITY_OPERATORS, tf.autograph.experimental.Feature.BUILTIN_FUNCTIONS))
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments18/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments19/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_autograph_options=None)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments19/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments2/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(input_signature=(tf.TensorSpec(shape=[None], dtype=tf.float32),))
+def func(x):
+  return x
+
+if __name__ == '__main__':
+    number = tf.constant([1.0, 1.0])
+    func(number)

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments2/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments20/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_follow_type_hints=True)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments20/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments21/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_follow_type_hints=False)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments21/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments22/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function(experimental_follow_type_hints=None)
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments22/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments23/in/A.py

@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+@tf.function
+def func():
+  pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments23/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments24/in/A.py

@@ -0,0 +1,12 @@
+import tensorflow as tf
+
+
+@tf.function(input_signature=(tf.TensorSpec(shape=[None], dtype=tf.float32),), autograph=False)
+def func(x):
+  print('Tracing with', x)
+  return x
+
+
+if __name__ == '__main__':
+    number = tf.constant([1.0, 1.0])
+    func(number)

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments24/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments25/in/A.py

@@ -0,0 +1,11 @@
+import custom
+
+
+@custom.decorator(input_signature=None)
+def func(x):
+  print('Tracing with', x)
+  return x
+
+
+if __name__ == '__main__':
+    func(1)

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments25/in/custom.py

@@ -0,0 +1,11 @@
+def decorator(input_signature=None):
+
+  def decorated(inner_function):
+
+      def wrapper(*args, **kwargs):
+          result = function(*args, **kwargs)
+          return result
+
+      return decorated
+
+  return decorator

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments25/in/requirements.txt

@@ -0,0 +1 @@
+

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments26/in/A.py

@@ -0,0 +1,12 @@
+import custom
+import tensorflow as tf
+
+
+@custom.decorator(input_signature=None)
+@tf.function(autograph=False)
+def func():
+    pass
+
+
+if __name__ == '__main__':
+    func()

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments26/in/custom.py

@@ -0,0 +1,11 @@
+def decorator(input_signature=None):
+
+  def decorated(inner_function):
+
+      def wrapper(*args, **kwargs):
+          result = function(*args, **kwargs)
+          return result
+
+      return decorated
+
+  return decorator

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments26/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments27/in/A.py

@@ -0,0 +1,11 @@
+import tensorflow as tf
+
+
+@tf.function(autograph=False)
+@tf.function(jit_compile=True)
+def func(x):
+  return x
+
+
+if __name__ == '__main__':
+    func(tf.constant(1))

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments27/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments28/in/A.py

@@ -0,0 +1,13 @@
+import tensorflow as tf
+
+var = (tf.TensorSpec(shape=[None], dtype=tf.float32),)
+
+
+@tf.function(input_signature=var)
+def func(x):
+  return x
+
+
+if __name__ == '__main__':
+    number = tf.constant([1.0, 1.0])
+    func(number)

edu.cuny.hunter.hybridize.tests/resources/HybridizeFunction/testDecoratorArguments28/in/requirements.txt

@@ -0,0 +1 @@
+tensorflow==2.9.3