Then you might have to rely on TensorFlow print operations like tf.print to output intermediate values.