Model.save is not working with tpu ('2.12.0 version)

I’m facing an error in saving the model on tpu’s. Neither option is working,

tf.saved_model.SaveOptions
model.save

error: TypeError: Unable to serialize [1.0092909 0.9846923 1.0169896 1.1289359 1.1968342 0.94357634
1.1685131 0.876992 0.81297153 0.88062966 1.261375 0.93040705] to JSON. Unrecognized type <class ‘tensorflow.python.framework.ops.EagerTensor’>.

Hi @Abdur_Rehman_Khan ,

Kindly refer to this documentation. Additionally, please update to TensorFlow version 2.16 for more functionality .

Thank You !