pytorch
34075e2c - [ONNX] Fix the issue of converting empty list to sequence. (#58651) (#61558)

Commit
3 years ago
[ONNX] Fix the issue of converting empty list to sequence. (#58651) (#61558) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61558 When we construct an empty list by python list comprehension, we need to avoid converting the node without inputs to onnx::Concat in shape_type_inference.cpp and peephole.cpp because it will create an invalid Concat node which doesn't have inputs. In addition, update the code to avoid passing a Sequence input to an onnx::Cast node which doesn't accept Sequence data type as an input. Add tests for the validation as well. Test Plan: Imported from OSS Reviewed By: nikithamalgifb Differential Revision: D29767989 Pulled By: SplitInfinity fbshipit-source-id: f97f172ff20eebda4c3744c7a934df36716f12a2 Co-authored-by: fatcat-z <jiz@microsoft.com>
Author
Parents
Loading