/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | dataset_checkpoint_test.py | 52 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 …]
|
D | iterator_test.py | 774 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/ |
D | range_dataset_serialization_test.py | 43 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/ |
D | quantize_training.cc | 139 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/ |
D | grappler_item.cc | 41 item.save_op = save_op; in WithGraph() 96 if (!save_op.empty()) { in NodesToPreserve() 97 result.insert(NodeName(save_op)); in NodesToPreserve()
|
D | grappler_item.h | 60 string save_op; member
|
D | grappler_item_builder.cc | 568 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/ |
D | 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/ |
D | 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/compiler/tf2tensorrt/convert/ |
D | trt_optimization_pass.cc | 162 LOG(INFO) << "Save Op = " << item.save_op; in PrintDebugInfo()
|
/external/tensorflow/tensorflow/python/training/ |
D | saver.py | 87 def save_op(self, filename_tensor, saveables): member in BaseSaverBuilder 209 save = self.save_op(filename_tensor, saveables)
|
D | saver_test.py | 182 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/ |
D | tf_op_files.txt | 215 tensorflow/core/kernels/save_op.cc
|
/external/tensorflow/tensorflow/core/kernels/ |
D | BUILD | 2804 ":save_op", 2871 name = "save_op", 2872 prefix = "save_op", 5801 "save_op.cc",
|