pytorch
722af775 - [ONNX] ConstantMap setters to update existing value instead of emplace (#67630) (#67812)

Commit
3 years ago
[ONNX] ConstantMap setters to update existing value instead of emplace (#67630) (#67812) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67812 `UpdateShape` uses `.emplace(tensorName, shapeValue)`. This will not update `shapeValue` for `tensorName`, if such name already exist in the map. Hence our code is not able to correct the shape inference error, even if we inferred the shape correctly later. Test Plan: Imported from OSS Reviewed By: msaroufim Differential Revision: D32181300 Pulled By: malfet fbshipit-source-id: 05c58ad3fdac683ad957996acde8f0ed6341781d Co-authored-by: BowenBao <bowbao@microsoft.com>
Author
Parents
Loading