Skip to content

Commit e3d7ff4

Browse files
authored
Merge pull request tensorflow#3106 from dakshvar22/master
Fix Issue : Change exporter.py in object_detection
2 parents 364f96d + 9ccb662 commit e3d7ff4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

research/object_detection/exporter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos(
6969
if optimize_graph:
7070
logging.info('Graph Rewriter optimizations enabled')
7171
rewrite_options = rewriter_config_pb2.RewriterConfig(
72-
optimize_tensor_layout=True)
72+
layout_optimizer=rewriter_config_pb2.RewriterConfig.ON)
7373
rewrite_options.optimizers.append('pruning')
7474
rewrite_options.optimizers.append('constfold')
7575
rewrite_options.optimizers.append('layout')

0 commit comments

Comments
 (0)