[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>