Replace print with tf.print to prevent side effects in tf.function

[allrob23]

Description:

This PR updates tf_func and tf_func2 by replacing all Python print statements with tf.print. Since these functions are decorated with @relaxed_tf_function (a wrapper for TensorFlow’s tf.function), using standard print statements can lead to unexpected behavior due to TensorFlow’s tracing mechanism. According to TensorFlow’s documentation:

Side effects, like printing, […] can behave unexpectedly inside a tf.function, sometimes executing twice or not at all. […] TensorFlow APIs like tf.print […] are the best way to ensure your code will be executed by the TensorFlow runtime with each call. (Source)

Changes Made:

• Replaced all print statements with tf.print.

This ensures that debug outputs execute consistently and aligns with TensorFlow’s best practices for side-effect management within tf.function.