Can't determine whether an API call is from TensorFlow when more than two functions are in a call chain

[tatianacv]

When processing the following:

import tensorflow as tf

def func2(t):
  pass

@tf.function
def func():
  a = tf.constant([[1.0, 2.0], [3.0, 4.0]])
  b = tf.constant([[1.0, 1.0], [0.0, 1.0]])
  c = tf.matmul(a, b)
  tensor = tf.Tensor(c.op, 0, tf.float32)
  func2(tensor)

func()

We found an error in determining whether we are dealing with a TensorFlow API. We found that the tensor analysis is empty:

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

Line 82 in 93c96d8

TensorTypeAnalysis analysis = E.performAnalysis(builder);

[khatchad]

@tatianacv Can you open a PR with the failing test?

[khatchad]

@tatianacv Can you please add a PR like this one to https://github.com/wala/ML/pulls?

[khatchad]

@tatianacv Can you please add a PR like this one to https://github.com/wala/ML/pulls?

Sorry, I know I asked for this, but I think we can use ponder-lab#26 for now.

[khatchad]

The passing test case for this issue is here:

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

Line 97 in 7e18125

testTf2("tf2n.py", "func2", 1, 4, 2);

Input file:

ML/com.ibm.wala.cast.python.test/data/tf2n.py

Lines 1 to 14 in 7e18125

	import tensorflow as tf
	def func2(t):
	pass
	@tf.function
	def func():
	a = tf.constant([[1.0, 2.0], [3.0, 4.0]])
	b = tf.constant([[1.0, 1.0], [0.0, 1.0]])
	c = tf.matmul(a, b)
	tensor = tf.Tensor(c.op, 0, tf.float32)
	func2(tensor)
	func()