pytorch
84190daf - [ONNX] Update instance_norm implementation and support training (#60538) (#64375)

Commit
4 years ago
[ONNX] Update instance_norm implementation and support training (#60538) (#64375) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64375 * Update the instance_norm track_running_stats=True implementation and support the training mode * Reference: https://github.com/pytorch/pytorch/blob/9baf75c86edc7f6cd1c04bf9f42d18bc0d05f504/aten/src/ATen/native/Normalization.cpp#L532 * Fix https://github.com/pytorch/pytorch/issues/53887 Test Plan: Imported from OSS Reviewed By: jansel Differential Revision: D30919605 Pulled By: malfet fbshipit-source-id: 306eb2a1122bb5d90dcb7c18260a3a2057a21c34 Co-authored-by: hwangdeyu <dejack953@outlook.com>
Author
Parents
Loading