()
| 1019 | |
| 1020 | |
| 1021 | def GetOptimizedGraph(): |
| 1022 | mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph()) |
| 1023 | config = config_pb2.ConfigProto() |
| 1024 | config.graph_options.rewrite_options.CopyFrom( |
| 1025 | rewriter_config_pb2.RewriterConfig( |
| 1026 | constant_folding=rewriter_config_pb2.RewriterConfig.OFF, |
| 1027 | memory_optimization=rewriter_config_pb2.RewriterConfig.MANUAL)) |
| 1028 | return tf_optimizer.OptimizeGraph(config, mg) |
| 1029 | |
| 1030 | |
| 1031 | if __name__ == "__main__": |
no test coverage detected