[r2.17-rocm-enhanced] Fix tf_keras#2710

jayfurmanek merged 1 commit intor2.17-rocm-enhancedROCm/tensorflow-upstream:r2.17-rocm-enhancedfrom

r2.17-rocm-enhanced-fix-tf-kerasROCm/tensorflow-upstream:r2.17-rocm-enhanced-fix-tf-kerasCopy head branch name to clipboard