We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 364f96d + 9ccb662 commit e3d7ff4Copy full SHA for e3d7ff4
research/object_detection/exporter.py
@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos(
69
if optimize_graph:
70
logging.info('Graph Rewriter optimizations enabled')
71
rewrite_options = rewriter_config_pb2.RewriterConfig(
72
- optimize_tensor_layout=True)
+ layout_optimizer=rewriter_config_pb2.RewriterConfig.ON)
73
rewrite_options.optimizers.append('pruning')
74
rewrite_options.optimizers.append('constfold')
75
rewrite_options.optimizers.append('layout')
0 commit comments