pytorch
fd756caa - [ONNX] Support nn.init.normal (#84149)

Commit
2 years ago
[ONNX] Support nn.init.normal (#84149) * Updated symbolic function for `aten::normal` to support additional generator arguments emitted from https://github.com/pytorch/pytorch/blob/5563248b5882231cb99105b042cc32bddd18b912/torch/csrc/jit/passes/remove_mutation.cpp#L51 * Added symbolic function for `aten::is_pinned` and `prim::layout`. Both are unused by ONNX later on. Fixes #83647 Pull Request resolved: https://github.com/pytorch/pytorch/pull/84149 Approved by: https://github.com/AllenTiTaiWang, https://github.com/abock
Author
Committer
Parents
Loading