onnxruntime/orttraining/orttraining/python/training/utils
..
data
hooks
__init__.py
ptable.py
torch_io_helper.py
torch_profile_utils.py
torch_type_map.py