Creates a tf.compat.v1.ConfigProto for configuring Grappler. Args: optimizers_list: List of strings that represents the list of optimizers. Returns: tf.ConfigProto.
(optimizers_list)
| 155 | |
| 156 | |
| 157 | def get_grappler_config(optimizers_list): |
| 158 | """Creates a tf.compat.v1.ConfigProto for configuring Grappler. |
| 159 | |
| 160 | Args: |
| 161 | optimizers_list: List of strings that represents the list of optimizers. |
| 162 | |
| 163 | Returns: |
| 164 | tf.ConfigProto. |
| 165 | """ |
| 166 | config = _config_pb2.ConfigProto() |
| 167 | rewrite_options = config.graph_options.rewrite_options |
| 168 | for optimizer in optimizers_list: |
| 169 | rewrite_options.optimizers.append(optimizer) |
| 170 | return config |
| 171 | |
| 172 | |
| 173 | def run_graph_optimizations(graph_def, |