tf.compat.v1.disable_eager_execution

Отключает немедленное выполнение.

tf.compat.v1.disable_eager_execution()

Эта функция может быть вызвана только до создания любых графов, операций или тензоров. Она может использоваться в начале программы для сложных проектов миграции с TensorFlow 1.x на 2.x.