Да, ваш код должен работать без каких-либо изменений.
Просто помните, что tf.keras - это конкретная реализация спецификации Keras API, но высокоуровневый API всегда одинаков. Могут быть некоторые различия между tf.keras и keras, но, возможно, какой-то дополнительный метод добавлен в базовый API, без изменений или удаления изменений.
Следовательно, если ваш код использует тольковысокоуровневый API Keras, без каких-либо дополнительных операций с бэкэндом, тогда вы можете быть уверены, что замена keras на tf.keras ничего не сломает.