Searched refs:restore_ops (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 112 restore_ops = {}
115 restore_ops[saveable.name] = saveable.restore(
117 return restore_ops
202 restore_ops = self.restore(file_prefix)
204 with ops.control_dependencies(restore_ops.values()):
316 restore_ops = {}
321 restore_ops.update(saver.restore(file_prefix, options))
323 return restore_ops
333 restore_ops = restore_fn()
338 for name, op in restore_ops.items():
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 253 …restore_ops = trackable._restore_from_checkpoint_position(self) # pylint: disable=protected-access
254 if restore_ops:
255 self._checkpoint.new_restore_ops(restore_ops)
442 def restore_ops(self): member in CheckpointPosition
452 (restore_ops, tensor_saveables,
454 restore_ops.extend(
456 return restore_ops
958 restore_ops = []
968 restore_ops.extend(new_restore_ops)
971 restore_ops.extend(
[all …]
|
| D | util.py | 228 self.restore_ops = []
271 self.restore_ops.extend(new_ops)
287 restore_ops = []
311 restore_ops.append(restore_op)
314 return restore_ops
839 session.run(self._checkpoint.restore_ops, feed_dict=self._feed_dict)
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | BUILD | 20 name = "restore_ops",
22 "restore_ops.cc",
25 "restore_ops.h",
71 ":restore_ops",
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 502 restore_ops = position.restore_ops()
503 if restore_ops:
505 if len(restore_ops) == 1:
506 obj._initializer_op = restore_ops[0]
508 obj._initializer_op = control_flow_ops.group(*restore_ops)
512 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, restore_ops)
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | BUILD | 162 "//tensorflow/c/experimental/saved_model/core/ops:restore_ops",
|