message: Store which class was patched
  Summary:
  Store the infromation about which class was patched to allow
  more introspection.
  Ref T57251
  Reviewers: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, christiana
  Maniphest Tasks: T57251
diff: @@ -98,6 +98,7 @@ class KerasExtensions:
  # Change the class of the instance to be the patched class.
  instance.__class__ = patched_cls
  + instance.__original_class__ = instance_cls
  # Call the extension constructor as it has not been called yet on this
  # instance.
  