| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | range_dataset_serialization_test.py | 48 save_op = io_ops.write_file(
51 return save_op
70 save_op = self._save_op(iterator._iterator_resource)
72 return init_op, get_next, save_op, restore_op
79 init_op, get_next, save_op, _ = _build_graph(start, stop)
85 self.evaluate(save_op)
99 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
105 self.evaluate(save_op)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | checkpoint_test.py | 63 save_op = io_ops.write_file(
66 return save_op
83 save_op = self._save_op(iterator._iterator_resource)
85 return init_op, get_next, save_op, restore_op
92 init_op, get_next, save_op, _ = _build_graph(start, stop)
98 sess.run(save_op)
112 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
118 sess.run(save_op)
137 save_op = self._save_op(iterator._iterator_resource)
139 return init_op, get_next, save_op, restore_op
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.cc | 137 Status FindSaveOp(const Graph* graph, Node** save_op, in FindSaveOp() argument
146 *save_op = node; in FindSaveOp()
183 Status ConnectVariablesToSaveOp(Graph* graph, Node* save_op, in ConnectVariablesToSaveOp() argument
202 NodeBuilder(save_op->name(), save_op->type_string()); in ConnectVariablesToSaveOp()
235 for (const Edge* edge : save_op->out_edges()) { in ConnectVariablesToSaveOp()
238 graph->RemoveNode(save_op); in ConnectVariablesToSaveOp()
248 Status AddRestoreVariableSubgraphs(Graph* graph, Node* save_op, in AddRestoreVariableSubgraphs() argument
252 StringPiece name_prefix = GetNodeNamePrefix(save_op); in AddRestoreVariableSubgraphs()
316 Node* save_op = nullptr; in AddSaveAndRestore() local
319 TF_RETURN_IF_ERROR(FindSaveOp(graph, &save_op, &in_edges, &found)); in AddSaveAndRestore()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 60 item.save_op = save_op; in WithGraph()
122 if (!save_op.empty()) { in NodesToPreserve()
123 result.insert(NodeName(save_op)); in NodesToPreserve()
|
| D | grappler_item.h | 61 string save_op; member
|
| D | grappler_item_builder.cc | 601 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
610 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 193 save_op = self.save(file_prefix)
195 with ops.control_dependencies([save_op]):
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.cc | 172 LOG(INFO) << "Save Op = " << item.save_op; in PrintDebugInfo()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 85 def save_op(self, filename_tensor, saveables): member in BaseSaverBuilder
205 save = self.save_op(filename_tensor, saveables)
|
| D | saver_test.py | 177 save_op = None
180 save_op = op
182 assert save_op is not None
183 for save_inp in save_op.inputs[3:]:
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 1162 save_op = saver.save(file_prefix, options=options)
1164 with ops.control_dependencies([save_op]):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 2980 ":save_op",
3047 name = "save_op",
3048 prefix = "save_op",
6192 "save_op.cc",
|