MacOS M2 upgrade Sonoma 14.0 can not train model with tensorflow and karas

I can train a yolov3 at MacOS M2 ventura with tensorflow-macos=2.9.0(karas=2.9.0) and tensorflow-mental=0.5.
But when I upgrade the system to Sonoma14.0. I can not train model with below error.’

`MPSKernel MTLComputePipelineStateCache unable to load function ndArrayConvolution2DGradientWithWeightsA14. Compute function exceeds available temporary registers: (null)

I also try upgrade the tensorflow-mac to 2.13 and metal to 1.0.0. The same code on M1 somoma system can work well. This is only happen on M2. Is there any one know the this?