Fix #20322 — Weight save/load fails when model uses a custom subclass of a built-in layer (e.g. LSTM)

Fix #20322 — Weight save/load fails when model uses a custom subclass of a built-in layer (e.g. LSTM) #10756