Closed chunnienc closed 1 month ago
The decomposition for aten.native_group_norm runs precision issue with large weight and bias. Therefore the NHWC rewriter is disabled (inserts additional transposes) for native_group_norm with weight and bias input tensors.
BUG=https://b.corp.google.com/issues/353734124, https://b.corp.google.com/issues/354780253
The decomposition for aten.native_group_norm runs precision issue with large weight and bias. Therefore the NHWC rewriter is disabled (inserts additional transposes) for native_group_norm with weight and bias input tensors.
BUG=https://b.corp.google.com/issues/353734124, https://b.corp.google.com/issues/354780253