Run a few passes on the GraphModule to propagate the dynamic shape information. Mainly used to provide dynamic shape information for delegation.
(
dynamic_memory_planning_mode: DynamicMemoryPlanningMode = DynamicMemoryPlanningMode.UPPER_BOUND,
)
| 518 | |
| 519 | |
| 520 | def propagate_dynamic_shape( |
| 521 | dynamic_memory_planning_mode: DynamicMemoryPlanningMode = DynamicMemoryPlanningMode.UPPER_BOUND, |
| 522 | ) -> List[PassType]: |
| 523 | """ |
| 524 | Run a few passes on the GraphModule to propagate the dynamic shape information. |
| 525 | |
| 526 | Mainly used to provide dynamic shape information for delegation. |
| 527 | """ |
| 528 | return [ |
| 529 | SpecPropPass(), |
| 530 | HintBasedSymShapeEvalPass(), |
| 531 | ] |