To improve performance on project of mine, i\'ve coded a function using tf.function to replace a function witch does not use tf. The result is that plain python code runs much (