Searched refs:saveables_cache (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | save_util_v1.py | 97 call_with_mapped_captures, saveables_cache): argument
115 call_with_mapped_captures, saveables_cache))
136 call_with_mapped_captures, saveables_cache): argument
139 if saveables_cache is None:
151 if saveables_cache is not None:
152 cached_attributes = saveables_cache.setdefault(object_to_save, {})
198 assert saveables_cache is None
251 saveables_cache=None): argument
277 saveables_cache=saveables_cache))
285 def serialize_object_graph_with_registered_savers(graph_view, saveables_cache): argument
[all …]
|
| D | restore.py | 222 saveables_cache = self._checkpoint.saveables_cache.setdefault(
224 if saveable_name in saveables_cache:
225 return [], {saveable_name: saveables_cache[saveable_name]}
230 saveables_cache[saveable_name] = saveable
276 saveables_cache = self._checkpoint.saveables_cache
277 if saveables_cache is None:
284 saveable_list = saveables_cache.get(self.trackable,
300 del saveables_cache[self.trackable]
315 if saveables_cache is not None:
316 saveables_cache.setdefault(self.trackable,
|
| D | graph_view.py | 130 def serialize_object_graph(self, saveables_cache=None): argument
158 self, saveables_cache))
|
| D | checkpoint.py | 223 restore_op_cache, graph_view, options, saveables_cache): argument
309 self.saveables_cache = saveables_cache
1186 saveables_cache=self._saveables_cache))
1490 saveables_cache=self._saveables_cache)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.tracking.-object-graph-view.pbtxt | 40 …argspec: "args=[\'self\', \'saveables_cache\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | d_checkpoint.py | 335 saveables_cache=self._saveables_cache)
|