db955d02720a415605a4060c8f41222ae06d9065,keras/saving/saved_model/save_impl.py,,wrap_layer_objects,#Any#Any#,81

Before Change

  layer_metrics = data_structures.wrap_or_unwrap(
      {m.name: m for m in layer._metrics})  // pylint: disable=protected-access
  return dict(
      variables=data_structures.wrap_or_unwrap(layer.variables),
      trainable_variables=data_structures.wrap_or_unwrap(
          layer.trainable_variables),
      non_trainable_variables=data_structures.wrap_or_unwrap(
          layer.non_trainable_variables),
      layers=data_structures.wrap_or_unwrap(utils.list_all_layers(layer)),
      metrics=data_structures.wrap_or_unwrap(layer.metrics),
      regularization_losses=data_structures.wrap_or_unwrap(
          wrapped_loss_functions),
      layer_regularization_losses=data_structures.wrap_or_unwrap(
          wrapped_layer_losses),
      layer_metrics=layer_metrics)
  // pylint: disable=protected-access


def wrap_layer_functions(layer, serialization_cache):

After Change

  layer_metrics = tf.__internal__.tracking.wrap(
      {m.name: m for m in layer._metrics})  // pylint: disable=protected-access
  return dict(
      variables=tf.__internal__.tracking.wrap(layer.variables),
      trainable_variables=tf.__internal__.tracking.wrap(
          layer.trainable_variables),
      non_trainable_variables=tf.__internal__.tracking.wrap(
          layer.non_trainable_variables),
      layers=tf.__internal__.tracking.wrap(utils.list_all_layers(layer)),
      metrics=tf.__internal__.tracking.wrap(layer.metrics),
      regularization_losses=tf.__internal__.tracking.wrap(
          wrapped_loss_functions),
      layer_regularization_losses=tf.__internal__.tracking.wrap(
          wrapped_layer_losses),
      layer_metrics=layer_metrics)
  // pylint: disable=protected-access


def wrap_layer_functions(layer, serialization_cache):

Instances