Searched refs:serialization_cache (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
D | layer_serialization.py | 72 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 …]
|
D | base_serialization.py | 57 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
|
D | model_serialization.py | 46 def _get_serialized_attributes_internal(self, serialization_cache): argument 51 if len(serialization_cache[constants.KERAS_CACHE_KEY]) == 1: 58 serialization_cache))
|
D | save_impl.py | 82 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/ |
D | base.py | 1043 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/ |
D | loss_scale_optimizer.py | 173 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/ |
D | base_layer_v1.py | 2342 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))
|
D | base_layer.py | 3102 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))
|
D | training.py | 2661 def _list_functions_for_serialization(self, serialization_cache): argument 2670 Model, self)._list_functions_for_serialization(serialization_cache)
|