Home
last modified time | relevance | path

Searched refs:serialization_cache (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/saved_model/
H A Dlayer_serialization.py68 def objects_to_serialize(self, serialization_cache): argument
70 serialization_cache).objects_to_serialize)
72 def functions_to_serialize(self, serialization_cache): argument
74 serialization_cache).functions_to_serialize)
76 def _get_serialized_attributes(self, serialization_cache): argument
78 keras_cache = serialization_cache.setdefault(constants.KERAS_CACHE_KEY, {})
90 serialization_cache)
96 def _get_serialized_attributes_internal(self, serialization_cache): argument
98 objects = save_impl.wrap_layer_objects(self.obj, serialization_cache)
99 functions = save_impl.wrap_layer_functions(self.obj, serialization_cache)
[all …]
H A Dbase_serialization.py50 def trackable_children(self, serialization_cache): argument
55 children = self.objects_to_serialize(serialization_cache)
56 children.update(self.functions_to_serialize(serialization_cache))
71 def objects_to_serialize(self, serialization_cache): argument
87 def functions_to_serialize(self, serialization_cache): argument
H A Dsave_impl.py82 def wrap_layer_objects(layer, serialization_cache): argument
103 keras_loss_cache = serialization_cache.setdefault('keras_losses', {})
133 def wrap_layer_functions(layer, serialization_cache): argument
154 original_fns = _replace_child_layer_functions(layer, serialization_cache)
221 def _replace_child_layer_functions(layer, serialization_cache): argument
282 if child_layer not in serialization_cache[constants.KERAS_CACHE_KEY]:
285 serialization_cache).functions)
288 serialization_cache[constants.KERAS_CACHE_KEY][child_layer].functions)
H A Dmodel_serialization.py42 def _get_serialized_attributes_internal(self, serialization_cache): argument
47 if len(serialization_cache[constants.KERAS_CACHE_KEY]) == 1:
54 serialization_cache))