flax
Use `.input_formats` and `.output_formats` in place of `.input_layouts` and `.output_layouts` respectively.
#4784
Merged
Merged