CodexBloom - AI-Powered Q&A Platform

Unexpected slow training performance when using tf.function with TensorFlow 2.10 and TPU

👀 Views: 0 💬 Answers: 1 📅 Created: 2025-06-14
tensorflow tpus tf.function performance

I'm experiencing a significant slowdown in my model training when I wrap my training step in a `tf.function` decorator while using a TPU for acceleration in Ten...