feat: support group_norm, batch_norm, and layer_norm by zewenli98 · Pull Request #2330

feat: support group_norm, batch_norm, and layer_norm by zewenli98 · Pull Request #2330 · pytorch/TensorRT

Conversation

Comment on lines +50 to +60

Comment on lines +117 to +121

Comment on lines +183 to +187

Comment on lines +173 to +174

Comment on lines +351 to +374

zewenli98 changed the title ~~fix: update batch_norm and layer_norm~~ feat: support group_norm, batch_norm, and layer_norm

Work is in progress, pull request should not be merged yet

label

support group norm, and improve batch and layer norms

gs-olive pushed a commit that referenced this pull request

This was referenced

Oct 10, 2023