Return the ConfigProto with all runtime deltas applied.
(self)
| 796 | |
| 797 | @property |
| 798 | def config(self): |
| 799 | """Return the ConfigProto with all runtime deltas applied.""" |
| 800 | # Ensure physical devices have been discovered and config has been imported |
| 801 | self._initialize_physical_devices() |
| 802 | |
| 803 | config = config_pb2.ConfigProto() |
| 804 | if self._config is not None: |
| 805 | config.CopyFrom(self._config) |
| 806 | |
| 807 | if self._optimizer_jit is not None: |
| 808 | config.graph_options.optimizer_options.global_jit_level = ( |
| 809 | config_pb2.OptimizerOptions.ON_1 |
| 810 | if self._optimizer_jit else config_pb2.OptimizerOptions.OFF) |
| 811 | if self._intra_op_parallelism_threads is not None: |
| 812 | config.intra_op_parallelism_threads = self._intra_op_parallelism_threads |
| 813 | if self._inter_op_parallelism_threads is not None: |
| 814 | config.inter_op_parallelism_threads = self._inter_op_parallelism_threads |
| 815 | |
| 816 | if self._soft_device_placement is not None: |
| 817 | config.allow_soft_placement = self._soft_device_placement |
| 818 | else: |
| 819 | config.allow_soft_placement = self.executing_eagerly() |
| 820 | |
| 821 | if self._log_device_placement is not None: |
| 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) |
| 836 | if toggle is None: |
| 837 | return |
| 838 | |
| 839 | setattr(config.graph_options.rewrite_options, |
| 840 | option, |
| 841 | toggle) |
| 842 | |
| 843 | rewriter_toggle("layout_optimizer") |
| 844 | rewriter_toggle("constant_folding") |
| 845 | rewriter_toggle("shape_optimization") |
| 846 | rewriter_toggle("remapping") |
| 847 | rewriter_toggle("arithmetic_optimization") |
| 848 | rewriter_toggle("dependency_optimization") |
| 849 | rewriter_toggle("loop_optimization") |
| 850 | rewriter_toggle("function_optimization") |
| 851 | rewriter_toggle("debug_stripper") |
| 852 | rewriter_bool("disable_model_pruning") |
| 853 | rewriter_toggle("scoped_allocator_optimization") |
| 854 | rewriter_toggle("pin_to_host_optimization") |
| 855 | rewriter_toggle("implementation_selector") |
nothing calls this directly
no test coverage detected