You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by "Ubospica (via GitHub)" <gi...@apache.org> on 2023/04/19 02:08:35 UTC

[GitHub] [tvm] Ubospica commented on a diff in pull request #14659: [Unity] Update docs for operators

Ubospica commented on code in PR #14659:
URL: https://github.com/apache/tvm/pull/14659#discussion_r1170732245


##########
python/tvm/relax/op/nn/nn.py:
##########
@@ -659,14 +659,15 @@ def batch_norm(
     beta: Expr,
     moving_mean: Expr,
     moving_var: Expr,
-    axis: int,
+    axis: int = 1,

Review Comment:
   Common batch norm operators takes the axis indexed 1 as the axis to calculate mean and variance, such as torch.nn.functional.batch_norm.



-- 
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: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org