-
Notifications
You must be signed in to change notification settings - Fork 109
[torchlib] aten.instance_norm #1280
Copy link
Copy link
Closed
Labels
contribution welcomeWe welcome code contributions for thisWe welcome code contributions for thisgood first issueGood for newcomersGood for newcomersmodule: torchlibRelated to the torch/aten function lib in developmentRelated to the torch/aten function lib in development
Metadata
Metadata
Assignees
Labels
contribution welcomeWe welcome code contributions for thisWe welcome code contributions for thisgood first issueGood for newcomersGood for newcomersmodule: torchlibRelated to the torch/aten function lib in developmentRelated to the torch/aten function lib in development
Related to this issue: #1262
We discovered that
aten::instance_normis decomposed during export, emitting combination of reshape + reduce + batchnorm.It is possible to skip decomposition in exporter, and try to convert to onnx::InstanceNormalization instead. This needs torchlib support for this operator.
cc @xiaowuhu, @fatcat-z, @justinchuby, @titaiwangms