| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| range_dataset_op_test.py | 177 save_op = io_ops.write_file(
180 return save_op
196 save_op = self._save_op(iterator._iterator_resource)
198 return init_op, get_next, save_op, restore_op
205 init_op, get_next, save_op, _ = _build_graph(start, stop)
211 sess.run(save_op)
225 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
231 sess.run(save_op)
245 save_op = self._save_op(iterator._iterator_resource)
247 return init_op, get_next, save_op, restore_o
[all...] |
| reader_dataset_ops_test.py | 299 save_op = io_ops.write_file(
302 return save_op
319 save_op = self._save_op(iterator._iterator_resource)
321 return init_op, get_next_op, save_op, restore_op
338 init_op, get_next_op, save_op, restore_op = self._build_iterator_graph(
351 sess.run(save_op)
365 init_op, get_next_op, save_op, restore_op = self._build_iterator_graph(
391 init_op, get_next_op, save_op, restore_op = self._build_iterator_graph(
404 sess.run(save_op)
418 init_op, get_next_op, save_op, restore_op = self._build_iterator_graph
[all...] |
| iterator_ops_test.py | 588 save_op = io_ops.write_file(
590 return save_op
606 save_op = _save_op(iterator._iterator_resource)
608 return init_op, get_next, save_op, restore_op
616 save_op = _save_op(iterator._iterator_resource)
618 return init_op, get_next_op, save_op, restore_op
622 init_op, _, save_op, _ = _build_range_dataset_graph()
625 sess.run(save_op)
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| range_dataset_op_test.py | 93 save_op = io_ops.write_file(
96 return save_op
113 save_op = self._save_op(iterator._iterator_resource)
115 return init_op, get_next, save_op, restore_op
122 init_op, get_next, save_op, _ = _build_graph(start, stop)
128 sess.run(save_op)
142 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
148 sess.run(save_op)
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item.cc | 36 save_op = other.save_op;
85 if (!save_op.empty()) {
86 result.insert(NodeName(save_op));
|
| grappler_item.h | 54 string save_op; member in struct:tensorflow::grappler::GrapplerItem
|
| grappler_item_builder.cc | 438 new_item->save_op = saver.save_tensor_name();
447 new_item->save_op = saver.save_tensor_name();
|
| /external/tensorflow/tensorflow/core/graph/ |
| quantize_training.cc | 139 Status FindSaveOp(const Graph* graph, Node** save_op,
148 *save_op = node;
158 // The restore_all op should have the same prefix of the save_op.
185 Status ConnectVariablesToSaveOp(Graph* graph, Node* save_op,
202 // Create a new save_op that has inputs to all the new variables.
204 NodeBuilder(save_op->name(), save_op->type_string());
233 // Remove the old save_op and add the new one.
237 for (const Edge* edge : save_op->out_edges()) {
240 graph->RemoveNode(save_op);
312 Node* save_op = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform_test.py | 162 save_op = graph_def.node.add()
163 save_op.name = save_name
164 save_op.op = 'NoOp'
188 save_op.attr['dtypes'].list.type[:] = types
199 meta_graph_transform._clean_save_and_restore(graph_def, save_op, ['/bar'])
200 self.assertEqual(save_op.attr['dtypes'].list.type[:], expected_types)
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| sequence_queueing_state_saver_test.py | 309 save_op = next_batch.save_state("state2", None)
311 save_op = next_batch.save_state("state1", np.random.rand(1, 1))
316 save_op = next_batch.save_state("state1", state_input)
317 sess.run([save_op],
|
| /external/tensorflow/tensorflow/python/training/ |
| saver.py | 205 def save_op(self, filename_tensor, saveables): member in class:BaseSaverBuilder
326 save = self.save_op(filename_tensor, saveables)
[all...] |
