Revise param_shapes to be a list of ordered dict (#1424)
* Revise param_shapes to be a list of ordered dict
* test i can push
* add tests; split z2 and z3 into separate funcs
Co-authored-by: Xiaopeng Li <xiaopel@amazon.com>
Co-authored-by: Stas Bekman <stas@stason.org>
Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
