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

Commit
3 years ago
[ONNX] Update slice process shape to support rank only inference (#65782) (#66149) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66149 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`. Test Plan: Imported from OSS Reviewed By: jansel Differential Revision: D31423840 Pulled By: malfet fbshipit-source-id: 17b2b24aa63435d5212ebe6bdf66ae3c348c4e3b Co-authored-by: BowenBao <bowbao@microsoft.com>
Author
Parents
Loading