| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | restore_ops.cc | 73 ImmediateOpPtr restore_op(ctx->CreateOperation()); in SingleRestore() local
74 TF_RETURN_IF_ERROR(restore_op->Reset("RestoreV2", "/cpu:0")); in SingleRestore()
75 TF_RETURN_IF_ERROR(restore_op->SetAttrTypeList("dtypes", &dtype, 1)); in SingleRestore()
92 TF_RETURN_IF_ERROR(restore_op->AddInput(prefix_handle.get())); in SingleRestore()
93 TF_RETURN_IF_ERROR(restore_op->AddInput(names_handle.get())); in SingleRestore()
94 TF_RETURN_IF_ERROR(restore_op->AddInput(shapes_and_slices_handle.get())); in SingleRestore()
98 TF_RETURN_IF_ERROR(restore_op->Execute( in SingleRestore()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | checkpoint_test.py | 71 restore_op = gen_dataset_ops.deserialize_iterator(iterator_resource,
73 return restore_op
84 restore_op = self._restore_op(iterator._iterator_resource)
85 return init_op, get_next, save_op, restore_op
101 init_op, get_next, _, restore_op = _build_graph(start, stop)
104 sess.run(restore_op)
112 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
120 sess.run(restore_op)
138 restore_op = self._restore_op(iterator._iterator_resource)
139 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 | 61 item.restore_op = restore_op; in WithGraph()
125 if (!restore_op.empty()) { in NodesToPreserve()
126 result.insert(NodeName(restore_op)); in NodesToPreserve()
|
| D | grappler_item.h | 62 string restore_op; member
|
| D | grappler_item_builder.cc | 602 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
611 new_item->restore_op = saver.restore_op_name(); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 426 restore_op = io_ops.restore_v2(
436 init_op = saveable_objects[0].restore([restore_op], restored_shapes=None)
440 restore_op.set_shape(variable.shape)
441 variable._initial_value = restore_op
|
| D | saver.py | 152 self.restore_op(filename_tensor, saveable, preferred_shard))
156 def restore_op(self, filename_tensor, saveable, preferred_shard): member in BaseSaverBuilder
509 restore_op = self._AddShardedRestoreOps(filename_tensor, per_device,
515 restore_op = self._AddRestoreOps(filename_tensor, saveables,
560 restore_op_name=restore_op.name,
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 182 restore_op = self._traced_restore(filename_tensor).op
186 restore_op_name=restore_op.name,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.cc | 292 Node* restore_op; in AddRestoreVariableSubgraphs() local
298 .Finalize(graph, &restore_op)); in AddRestoreVariableSubgraphs()
304 .Input(restore_op) in AddRestoreVariableSubgraphs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.cc | 173 LOG(INFO) << "Restore Op = " << item.restore_op; in PrintDebugInfo()
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 310 for name, restore_op in sorted(new_restore_ops.items()):
311 restore_ops.append(restore_op)
313 self.restore_ops_by_name[name] = restore_op
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 2979 ":restore_op",
3041 name = "restore_op",
3042 prefix = "restore_op",
6189 "restore_op.cc",
|