Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[MLIR][TORCH] Add support for dim=None to Aten[Var|Std]DimOp #1159

Merged
merged 1 commit into from
Aug 5, 2022
Merged

[MLIR][TORCH] Add support for dim=None to Aten[Var|Std]DimOp #1159

merged 1 commit into from
Aug 5, 2022

Commits on Aug 5, 2022

  1. [MLIR][TORCH] Add support for dim=None to Aten[Var|Std]DimOp 

    PyTorch recently added support for `dim=None` in the `torch.var`
(pytorch/pytorch@5ca9b2b)
and `torch.std`op (pytorch/pytorch@eb0e30e).
This commit adds the corresponding support in torch-mlir.

Signed-Off By: Vivek Khandelwal<[email protected]>
    @vivekkhandelwal1
    vivekkhandelwal1 committed Aug 5, 2022
    Configuration menu
    Copy the full SHA
    bd6b632 View commit details
    Browse the repository at this point in the history