pytorch

788ff062 - [decomp] fix decomp of batch_norm when weight/bias is not flattened (#101059)

[decomp] fix decomp of batch_norm when weight/bias is not flattened (#101059) Fix https://github.com/pytorch/pytorch/issues/100970 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101059 Approved by: https://github.com/ezyang