working on failing shape inference tests