Asuka0630 opened a new pull request, #18609: URL: https://github.com/apache/tvm/pull/18609
Dear reviewers, **Why** The previous implementation of batch_norm incorrectly conflated the computation of mean and variance between training and evaluation modes. Additionally, for '_native_batch_norm_legit.no_stats‘, using instance_norm to handle normalization ignored the batch dimension, leading to incorrect behavior. **How** This PR includes the following fixes: 1. Corrects the computation logic to properly distinguish between training and evaluation modes. 2. Ensures the batch dimension is properly accounted for in `_batch_norm_legit_no_stats`. **Environment** GPU: NVIDIA A100-SXM4-80GB -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
