| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | dataset_checkpoint_test.py | 60 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
62 return restore_op
72 restore_op = self._restore_op(iterator._iterator_resource)
73 return init_op, get_next, save_op, restore_op
89 init_op, get_next, _, restore_op = _build_graph(start, stop)
92 sess.run(restore_op)
100 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
107 sess.run(restore_op)
121 restore_op = self._restore_op(iterator._iterator_resource)
122 return init_op, get_next, save_op, restore_op
[all …]
|
| D | iterator_test.py | 781 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
783 return restore_op
793 restore_op = _restore_op(iterator._iterator_resource)
794 return init_op, get_next, save_op, restore_op
803 restore_op = _restore_op(iterator._iterator_resource)
804 return init_op, get_next_op, save_op, restore_op
819 _, _, _, restore_op = _build_reader_dataset_graph()
822 sess.run(restore_op)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | range_dataset_serialization_test.py | 52 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
54 return restore_op
64 restore_op = self._restore_op(iterator._iterator_resource)
65 return init_op, get_next, save_op, restore_op
81 init_op, get_next, _, restore_op = _build_graph(start, stop)
84 self.evaluate(restore_op)
92 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
99 self.evaluate(restore_op)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 312 restore_op = io_ops.restore_v2(
322 init_op = saveable_objects[0].restore([restore_op], restored_shapes=None)
326 restore_op.set_shape(variable.shape)
327 variable._initial_value = restore_op
|
| D | saver.py | 154 self.restore_op(filename_tensor, saveable, preferred_shard))
158 def restore_op(self, filename_tensor, saveable, preferred_shard): member in BaseSaverBuilder
505 restore_op = self._AddShardedRestoreOps(filename_tensor, per_device,
511 restore_op = self._AddRestoreOps(filename_tensor, saveables,
556 restore_op_name=restore_op.name,
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 42 item.restore_op = restore_op; in WithGraph()
99 if (!restore_op.empty()) { in NodesToPreserve()
100 result.insert(NodeName(restore_op)); in NodesToPreserve()
|
| D | grappler_item.h | 61 string restore_op; member
|
| D | grappler_item_builder.cc | 569 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
578 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 67 restore_op = def_function.function(
73 restore_op_name=restore_op.name,
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | checkpoint_utils.py | 119 restore_op = io_ops.restore_v2(
121 variable._initializer_op = state_ops.assign(variable, restore_op)
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 294 Node* restore_op; in AddRestoreVariableSubgraphs() local
300 .Finalize(graph, &restore_op)); in AddRestoreVariableSubgraphs()
306 .Input(restore_op) in AddRestoreVariableSubgraphs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.cc | 163 LOG(INFO) << "Restore Op = " << item.restore_op; in PrintDebugInfo()
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 165 for saveable, restore_op in zip(validated_saveables, new_restore_ops):
167 self.restore_ops_by_name[saveable.name] = restore_op
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 211 tensorflow/core/kernels/restore_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 2803 ":restore_op",
2865 name = "restore_op",
2866 prefix = "restore_op",
5799 "restore_op.cc",
|