(option)
| 822 | config.log_device_placement = self._log_device_placement |
| 823 | |
| 824 | def rewriter_toggle(option): |
| 825 | toggle = self._optimizer_experimental_options.get(option, None) |
| 826 | if toggle is None: |
| 827 | return |
| 828 | |
| 829 | setattr(config.graph_options.rewrite_options, |
| 830 | option, |
| 831 | (rewriter_config_pb2.RewriterConfig.ON |
| 832 | if toggle else rewriter_config_pb2.RewriterConfig.OFF)) |
| 833 | |
| 834 | def rewriter_bool(option): |
| 835 | toggle = self._optimizer_experimental_options.get(option, None) |