Searched refs:restore_ops (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | restore.py | 55 restore_ops = self._restore_descendants()
56 if restore_ops:
57 self._checkpoint.new_restore_ops(restore_ops)
322 def restore_ops(self): member in CheckpointPosition
335 (restore_ops, tensor_saveables, python_positions,
337 restore_ops.extend(
339 return restore_ops
437 restore_ops = []
447 restore_ops.extend(new_restore_ops)
457 restore_ops.extend(
[all …]
|
| D | functional_saver.py | 277 restore_ops = self.restore(file_prefix)
279 with ops.control_dependencies(restore_ops.values()):
414 restore_ops = {}
436 restore_ops.update(ret)
440 return restore_ops
456 restore_ops = tf_function_restore()
458 restore_ops = restore_fn()
460 return restore_ops
|
| D | checkpoint.py | 276 self.restore_ops = []
321 self.restore_ops.extend(new_ops)
341 restore_ops = []
356 restore_ops.append(restore_op)
359 return restore_ops
886 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/core/kernels/ |
| D | save_restore_tensor.cc | 373 std::vector<RestoreOp> restore_ops; in RestoreTensorsV2() local
374 restore_ops.reserve(tensor_names_flat.size()); in RestoreTensorsV2()
376 restore_ops.push_back({context, i, tensor_names_flat(i), in RestoreTensorsV2()
384 restore_ops, [](const RestoreOp& op) { return op.tensor_name; })); in RestoreTensorsV2()
387 for (const RestoreOp& restore_op : restore_ops) { in RestoreTensorsV2()
407 for (RestoreOp& restore_op : restore_ops) { in RestoreTensorsV2()
438 for (const RestoreOp& restore_op : restore_ops) { in RestoreTensorsV2()
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | d_checkpoint.py | 158 restore_ops = {}
161 restore_ops[saveable.name] = saveable.restore(
163 return restore_ops
194 restore_ops = []
219 restore_ops.append(restore_op)
222 return restore_ops
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 547 restore_ops = position.restore_ops()
548 if restore_ops:
550 if len(restore_ops) == 1:
551 obj._initializer_op = restore_ops[0]
553 obj._initializer_op = control_flow_ops.group(*restore_ops)
557 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, restore_ops)
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 812 restore_ops = {}
825 restore_ops[saveable.name] = saveable.restore(
827 return restore_ops
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | BUILD | 164 "//tensorflow/c/experimental/saved_model/core/ops:restore_ops",
|