onnxruntime
4f15b427 - Customize _get_tensor_rank for model export in stage3 (#18294)

Commit
2 years ago
Customize _get_tensor_rank for model export in stage3 (#18294) ### 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. -->
Author
Parents
Loading