Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dlayer_serialization.py72 def objects_to_serialize(self, serialization_cache): argument
74 serialization_cache).objects_to_serialize)
76 def functions_to_serialize(self, serialization_cache): argument
78 serialization_cache).functions_to_serialize)
80 def _get_serialized_attributes(self, serialization_cache): argument
82 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.py57 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
94 fns = self.functions_to_serialize(serialization_cache)
101 self.obj, serialization_cache))
116 def objects_to_serialize(self, serialization_cache): argument
132 def functions_to_serialize(self, serialization_cache): argument
Dmodel_serialization.py46 def _get_serialized_attributes_internal(self, serialization_cache): argument
51 if len(serialization_cache[constants.KERAS_CACHE_KEY]) == 1:
58 serialization_cache))
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)
209 def _replace_child_layer_functions(layer, serialization_cache): argument
270 if child_layer not in serialization_cache[constants.KERAS_CACHE_KEY]:
273 serialization_cache).functions)
276 serialization_cache[constants.KERAS_CACHE_KEY][child_layer].functions)
/external/tensorflow/tensorflow/python/training/tracking/
Dbase.py1043 def _list_extra_dependencies_for_serialization(self, serialization_cache): argument
1070 del serialization_cache
1073 def _list_functions_for_serialization(self, serialization_cache): argument
1088 del serialization_cache
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py173 def _list_extra_dependencies_for_serialization(self, serialization_cache): argument
175 serialization_cache)
177 def _list_functions_for_serialization(self, serialization_cache): argument
179 serialization_cache)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_v1.py2342 def _list_extra_dependencies_for_serialization(self, serialization_cache): argument
2344 .list_extra_dependencies_for_serialization(serialization_cache))
2346 def _list_functions_for_serialization(self, serialization_cache): argument
2348 .list_functions_for_serialization(serialization_cache))
Dbase_layer.py3102 def _list_extra_dependencies_for_serialization(self, serialization_cache): argument
3104 .list_extra_dependencies_for_serialization(serialization_cache))
3106 def _list_functions_for_serialization(self, serialization_cache): argument
3108 .list_functions_for_serialization(serialization_cache))
Dtraining.py2661 def _list_functions_for_serialization(self, serialization_cache): argument
2670 Model, self)._list_functions_for_serialization(serialization_cache)