mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-17 01:44:45 +00:00
### Customize _get_tensor_rank for model export in stage3 Weight/Params sizes are all (0), so exporter logic depending on input shape will fail. This PR override `_get_tensor_rank` function by retrieving the shape for weight differently. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> |
||
|---|---|---|
| .. | ||
| deprecated | ||
| training | ||
| checkpointing_utils.py | ||
| ort_trainer.py | ||
| orttraining_pybind_common.h | ||
| orttraining_pybind_state.cc | ||
| orttraining_python_module.cc | ||
| orttraining_python_module_eager.h | ||
| pt_patch.py | ||