search

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
682 stars 89 forks source link

About Batchnorm #91

Open ryujaehun opened 2 years ago

ryujaehun commented 2 years ago

Hi , TASO is only concerned about 2d batch norm case but some network has 1d or 3d batch norm case.

https://github.com/jiazhihao/TASO/blob/f11782c4db180ad7d1e7ff7f4bb82e574ac3b6b8/src/core/batchnorm.cc#L72

I have a question about the above code. If I modify the above code, will it work rightly?