mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-27 22:45:57 +00:00
fe0db63dee
### Upstream reshape of merging batch/sequence For Reshape node that fulfills following requirements: - input data rank = 3 - input shape is constant initializer, the untorched dim value MUST be a constant value. - Reshape is merging the first dimension, so output data rank = 2. We upstream it to make it run as earlier as possible. Doing this will allow us to upstream other operators (Gather) that is blocked by those kind of Reshape node. Currently, we did not enable it in graph_transformer_utils, since the combined upstream gather changes are not ready yet. Before:  After:  |
||
|---|---|---|
| .. | ||
| amp | ||
| api | ||
| experimental | ||
| onnxblock | ||
| optim | ||
| ortmodule | ||
| torchdynamo | ||
| utils | ||
| __init__.py | ||
| _checkpoint_storage.py | ||
| _utils.py | ||
| artifacts.py | ||
| checkpoint.py | ||
| model_desc_validation.py | ||
| orttrainer.py | ||
| orttrainer_options.py | ||
| postprocess.py | ||
