Parses input tensorflow Saver into SaverDef proto.
(input_saver, input_binary)
| 270 | |
| 271 | |
| 272 | def _parse_input_saver_proto(input_saver, input_binary): |
| 273 | """Parses input tensorflow Saver into SaverDef proto.""" |
| 274 | if not gfile.Exists(input_saver): |
| 275 | raise IOError("Input saver file '" + input_saver + "' does not exist!") |
| 276 | mode = "rb" if input_binary else "r" |
| 277 | with gfile.GFile(input_saver, mode) as f: |
| 278 | saver_def = saver_pb2.SaverDef() |
| 279 | if input_binary: |
| 280 | saver_def.ParseFromString(f.read()) |
| 281 | else: |
| 282 | text_format.Merge(f.read(), saver_def) |
| 283 | return saver_def |
| 284 | |
| 285 | |
| 286 | def freeze_graph(input_graph, |
no test coverage detected