Override the runtime configs of params from flags.
(self, params)
| 336 | return params |
| 337 | |
| 338 | def parse_runtime(self, params): |
| 339 | """Override the runtime configs of params from flags.""" |
| 340 | # Override the TPU address and tf.data service address. |
| 341 | params.override({ |
| 342 | 'runtime': { |
| 343 | 'tpu': self._flags_obj.tpu, |
| 344 | }, |
| 345 | }) |
| 346 | return params |
| 347 | |
| 348 | def parse_data_service(self, params): |
| 349 | """Override the data service configs of params from flags.""" |