The below code works fine with Tensorflow 1.15 but doesn’t work with the newer versions of Tensorflow like 2.9.1
from tensorflow.python.keras.utils.tf_utils import smart_cond
I want to upgrade my code to work with Tensorflow 2.9.1 and I need the smart_cond function. Does anyone know where can I find that function in the newer version?