fix regression and enable custom autograd func tests in CIs (#8868)
* fix regression and enable tests in CIs
* Update orttraining/orttraining/python/training/ortmodule/_custom_autograd_function.py
Co-authored-by: Wei-Sheng Chin <wschin@outlook.com>
* fix
Co-authored-by: Wei-Sheng Chin <wschin@outlook.com>