"python/vscode:/vscode.git/clone" did not exist on "8dbdc018a31e1437fb40fec48fd0a9a6ba4fdaee"
Unverified Commit 9ccb6622 authored by Daksh Varshneya's avatar Daksh Varshneya Committed by GitHub
Browse files

Make parameter value more generic and adaptable.

Incorporated the suggested changes.
parent da98beae
......@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos(
if optimize_graph:
logging.info('Graph Rewriter optimizations enabled')
rewrite_options = rewriter_config_pb2.RewriterConfig(
layout_optimizer=1)
layout_optimizer=rewriter_config_pb2.RewriterConfig.ON)
rewrite_options.optimizers.append('pruning')
rewrite_options.optimizers.append('constfold')
rewrite_options.optimizers.append('layout')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment