Add support for -hlo-flatten-computation-args-result for importing entry_computation_layout.
#40400
| Job | Run time |
|---|---|
| 16s | |
| 16s |