pytorch
6f08ddfc - [ONNX] Enable aten:normal op and add tests for aten:uniform op. (#60441) (#61560)

Commit
3 years ago
[ONNX] Enable aten:normal op and add tests for aten:uniform op. (#60441) (#61560) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61560 1. Add a new symbolic function broadcast_tensors() to support exporting torch.broadcast_tensors() function. This is required by exporting torch.distribution.normal() function. 2. Add a new symbolic function normal() to support exporting torch.distribution.normal() function. 3. Add relative tests for normal and uniform ops as well. Test Plan: Imported from OSS Reviewed By: nikithamalgifb Differential Revision: D29767995 Pulled By: SplitInfinity fbshipit-source-id: acfe5e7801d00c0df8ca46966bbd6015fed0045e Co-authored-by: Jay Zhang <jiz@microsoft.com>
Author
Parents
Loading