pytorch
ce48b140 - [ONNX] Improve index_put symbolic to handle singular Bool updates (#53690) (#54863)

Commit
4 years ago
[ONNX] Improve index_put symbolic to handle singular Bool updates (#53690) (#54863) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54863 Adds support for cases where the updates to the index_put node is a single Bool value, such as the case shown below ``` mask[indices] = True ``` Fixes #53507 Test Plan: Imported from OSS Reviewed By: nikithamalgifb Differential Revision: D27408977 Pulled By: SplitInfinity fbshipit-source-id: bcfb55b50ce76b3d4913ffbc16cdef1f98cb7a84
Parents
Loading