Add support for `-hlo-flatten-computation-args-result` for importing `entry_computation_layout`. · openxla/xla@18ec3a3