Searched refs:restore_op (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | checkpoint_test.py | 65 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
67 return restore_op
78 restore_op = self._restore_op(iterator._iterator_resource)
79 return init_op, get_next, save_op, restore_op
95 init_op, get_next, _, restore_op = _build_graph(start, stop)
98 sess.run(restore_op)
106 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
114 sess.run(restore_op)
132 restore_op = self._restore_op(iterator._iterator_resource)
133 return init_op, get_next, save_op, restore_op
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | range_dataset_serialization_test.py | 57 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
59 return restore_op
71 restore_op = self._restore_op(iterator._iterator_resource)
72 return init_op, get_next, save_op, restore_op
88 init_op, get_next, _, restore_op = _build_graph(start, stop)
91 self.evaluate(restore_op)
99 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
106 self.evaluate(restore_op)
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 43 item.restore_op = restore_op; in WithGraph()
100 if (!restore_op.empty()) { in NodesToPreserve()
101 result.insert(NodeName(restore_op)); in NodesToPreserve()
|
| D | grappler_item.h | 62 string restore_op; member
|
| D | grappler_item_builder.cc | 601 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
610 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 412 restore_op = io_ops.restore_v2(
422 init_op = saveable_objects[0].restore([restore_op], restored_shapes=None)
426 restore_op.set_shape(variable.shape)
427 variable._initial_value = restore_op
|
| D | saver.py | 153 self.restore_op(filename_tensor, saveable, preferred_shard))
157 def restore_op(self, filename_tensor, saveable, preferred_shard): member in BaseSaverBuilder
501 restore_op = self._AddShardedRestoreOps(filename_tensor, per_device,
507 restore_op = self._AddRestoreOps(filename_tensor, saveables,
552 restore_op_name=restore_op.name,
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 168 restore_op = self._traced_restore(filename_tensor).op
172 restore_op_name=restore_op.name,
|
| /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/python/training/tracking/ |
| D | util.py | 291 for name, restore_op in sorted(new_restore_ops.items()):
292 restore_ops.append(restore_op)
294 self.restore_ops_by_name[name] = restore_op
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.cc | 161 LOG(INFO) << "Restore Op = " << item.restore_op; in PrintDebugInfo()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3346 ":restore_op",
3408 name = "restore_op",
3409 prefix = "restore_op",
6669 "restore_op.cc",
|