pytorch
a572f70f - [ONNX] Support torch.isinf, torch.any and torch.all export to ONNX (#53328) (#53529)

Commit
3 years ago
[ONNX] Support torch.isinf, torch.any and torch.all export to ONNX (#53328) (#53529) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53529 Supported for ONNX export after opset 10. This is not exportable to opsets < 10 due to 1. onnx::IsInf is introduced in opset 10 2. onnx::Equal does not accept float tensor prior to opset 11 Test Plan: Imported from OSS Reviewed By: pbelevich, malfet Differential Revision: D26922418 Pulled By: SplitInfinity fbshipit-source-id: 69bcba50520fa3d69db4bd4c2b9f88c00146fca7
Author
Parents
Loading