Searched defs:save_op (Results 1 – 4 of 4) sorted by relevance
137 Status FindSaveOp(const Graph* graph, Node** save_op, in FindSaveOp()183 Status ConnectVariablesToSaveOp(Graph* graph, Node* save_op, in ConnectVariablesToSaveOp()248 Status AddRestoreVariableSubgraphs(Graph* graph, Node* save_op, in AddRestoreVariableSubgraphs()316 Node* save_op = nullptr; in AddSaveAndRestore() local
61 string save_op; member
128 absl::string_view save_op = def.save_tensor_name(); in SaveState() local
110 def save_op(self, filename_tensor, saveables): member in BaseSaverBuilder