I have observed the same behavior like you, as I mention in the post https://developer.apple.com/forums/thread/709142. I am using tensorflow-macos 2.8.0, tensorflow-metal 0.5.0  and  python 3.8.13.