Freeze the current graph. Args: session: Tensorflow sessions containing the graph outputs: List of output tensors Returns: The frozen graph_def.
(session, outputs)
| 270 | |
| 271 | |
| 272 | def freeze_graph(session, outputs): |
| 273 | """Freeze the current graph. |
| 274 | |
| 275 | Args: |
| 276 | session: Tensorflow sessions containing the graph |
| 277 | outputs: List of output tensors |
| 278 | |
| 279 | Returns: |
| 280 | The frozen graph_def. |
| 281 | """ |
| 282 | return tf_graph_util.convert_variables_to_constants( |
| 283 | session, session.graph.as_graph_def(), [x.op.name for x in outputs]) |
| 284 | |
| 285 | |
| 286 | @register_make_test_function() |
no test coverage detected