Home
last modified time | relevance | path

Searched refs:save_op (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/data/kernel_tests/
Ddataset_checkpoint_test.py52 save_op = io_ops.write_file(
55 return save_op
71 save_op = self._save_op(iterator._iterator_resource)
73 return init_op, get_next, save_op, restore_op
80 init_op, get_next, save_op, _ = _build_graph(start, stop)
86 sess.run(save_op)
100 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
106 sess.run(save_op)
120 save_op = self._save_op(iterator._iterator_resource)
122 return init_op, get_next, save_op, restore_op
[all …]
Diterator_test.py774 save_op = io_ops.write_file(
776 return save_op
792 save_op = _save_op(iterator._iterator_resource)
794 return init_op, get_next, save_op, restore_op
802 save_op = _save_op(iterator._iterator_resource)
804 return init_op, get_next_op, save_op, restore_op
808 init_op, _, save_op, _ = _build_range_dataset_graph()
811 sess.run(save_op)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Drange_dataset_serialization_test.py43 save_op = io_ops.write_file(
46 return save_op
63 save_op = self._save_op(iterator._iterator_resource)
65 return init_op, get_next, save_op, restore_op
72 init_op, get_next, save_op, _ = _build_graph(start, stop)
78 self.evaluate(save_op)
92 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
98 self.evaluate(save_op)
/external/tensorflow/tensorflow/core/graph/
Dquantize_training.cc139 Status FindSaveOp(const Graph* graph, Node** save_op, in FindSaveOp() argument
148 *save_op = node; in FindSaveOp()
185 Status ConnectVariablesToSaveOp(Graph* graph, Node* save_op, in ConnectVariablesToSaveOp() argument
204 NodeBuilder(save_op->name(), save_op->type_string()); in ConnectVariablesToSaveOp()
237 for (const Edge* edge : save_op->out_edges()) { in ConnectVariablesToSaveOp()
240 graph->RemoveNode(save_op); in ConnectVariablesToSaveOp()
250 Status AddRestoreVariableSubgraphs(Graph* graph, Node* save_op, in AddRestoreVariableSubgraphs() argument
254 StringPiece name_prefix = GetNodeNamePrefix(save_op); in AddRestoreVariableSubgraphs()
318 Node* save_op = nullptr; in AddSaveAndRestore() local
321 TF_RETURN_IF_ERROR(FindSaveOp(graph, &save_op, &in_edges, &found)); in AddSaveAndRestore()
[all …]
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc41 item.save_op = save_op; in WithGraph()
96 if (!save_op.empty()) { in NodesToPreserve()
97 result.insert(NodeName(save_op)); in NodesToPreserve()
Dgrappler_item.h60 string save_op; member
Dgrappler_item_builder.cc568 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
577 new_item->save_op = saver.save_tensor_name(); in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform_test.py162 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/
Dsequence_queueing_state_saver_test.py309 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/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc162 LOG(INFO) << "Save Op = " << item.save_op; in PrintDebugInfo()
/external/tensorflow/tensorflow/python/training/
Dsaver.py87 def save_op(self, filename_tensor, saveables): member in BaseSaverBuilder
209 save = self.save_op(filename_tensor, saveables)
Dsaver_test.py182 save_op = None
185 save_op = op
187 assert save_op is not None
188 for save_inp in save_op.inputs[3:]:
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt215 tensorflow/core/kernels/save_op.cc
/external/tensorflow/tensorflow/core/kernels/
DBUILD2804 ":save_op",
2871 name = "save_op",
2872 prefix = "save_op",
5801 "save_op.cc",