Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dlayer_serialization.py73 def objects_to_serialize(self, serialization_cache): argument
75 serialization_cache).objects_to_serialize)
77 def functions_to_serialize(self, serialization_cache): argument
79 serialization_cache).functions_to_serialize)
81 def _get_serialized_attributes(self, serialization_cache): argument
83 keras_cache = serialization_cache.setdefault(constants.KERAS_CACHE_KEY, {})
94 serialization_cache)
100 def _get_serialized_attributes_internal(self, serialization_cache): argument
102 objects = save_impl.wrap_layer_objects(self.obj, serialization_cache)
103 functions = save_impl.wrap_layer_functions(self.obj, serialization_cache)
[all …]
Dbase_serialization.py60 def list_extra_dependencies_for_serialization(self, serialization_cache): argument
78 return self.objects_to_serialize(serialization_cache)
80 def list_functions_for_serialization(self, serialization_cache): argument
91 fns = self.functions_to_serialize(serialization_cache)
98 self.obj, serialization_cache))
113 def objects_to_serialize(self, serialization_cache): argument
129 def functions_to_serialize(self, serialization_cache): argument
Dmodel_serialization.py41 def _get_serialized_attributes_internal(self, serialization_cache): argument
46 if len(serialization_cache[constants.KERAS_CACHE_KEY]) == 1:
53 serialization_cache))
Dsave_impl.py94 def wrap_layer_objects(layer, serialization_cache): argument
115 keras_loss_cache = serialization_cache.setdefault('keras_losses', {})
140 def wrap_layer_functions(layer, serialization_cache): argument
161 original_fns = _replace_child_layer_functions(layer, serialization_cache)
216 def _replace_child_layer_functions(layer, serialization_cache): argument
246 if child_layer not in serialization_cache[constants.KERAS_CACHE_KEY]:
249 serialization_cache).functions)
252 serialization_cache[constants.KERAS_CACHE_KEY][child_layer].functions)
/external/tensorflow/tensorflow/python/training/tracking/
Dbase.py977 def _list_extra_dependencies_for_serialization(self, serialization_cache): argument
1004 del serialization_cache
1007 def _list_functions_for_serialization(self, serialization_cache): argument
1022 del serialization_cache
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py2529 def _list_extra_dependencies_for_serialization(self, serialization_cache): argument
2531 .list_extra_dependencies_for_serialization(serialization_cache))
2533 def _list_functions_for_serialization(self, serialization_cache): argument
2535 .list_functions_for_serialization(serialization_cache))