Add support for -hlo-flatten-computation-args-result for importing entry_computation_layout.
#5715
| Job | Run time |
|---|---|
| 1m 0s | |
| 1m 0s |