pytorch
58092880 - [ONNX] Adding 'numel' and 'to' export for script module (#36501)

Commit
5 years ago
[ONNX] Adding 'numel' and 'to' export for script module (#36501) Summary: These two ops are needed for torchvision model export. Since we're scripting a part of the code for dynamic export of models (in https://github.com/pytorch/vision/pull/2052), these two changes are requited. Pull Request resolved: https://github.com/pytorch/pytorch/pull/36501 Reviewed By: hl475 Differential Revision: D21260721 Pulled By: houseroad fbshipit-source-id: 86d9d38665a4a36d22cec741012d976e5bd8d36b
Author
Parents
Loading