pytorch
f49334ce - [ONNX] Update slice process shape to support rank only inference (#65782)

Commit
3 years ago
[ONNX] Update slice process shape to support rank only inference (#65782) Updated logic will be able to infer rank of slice output, when only rank is known for slice input. Enables cases where `ConstantValueMap::HasRank(input)` is `True`, while `ConstantValueMap::HasShape(input)` is `False`. Co-authored-by: BowenBao <bowbao@microsoft.com> ghstack-source-id: 01477ba2671657fc432bf817f0317e3d1809402d Pull Request resolved: https://github.com/pytorch/pytorch/pull/66149
Author
Parents
Loading