pytorch

236565f4 - [functorch] add layer norm support, clean up some binary cross entropy support (pytorch/functorch#807)

[functorch] add layer norm support, clean up some binary cross entropy support (pytorch/functorch#807) * add layer norm support, clean up some binary cross entropy support * zero returns have the same shape as their input