mirror of https://github.com/llvm/torch-mlir
c681c3497a
This commit fixes the shape calculation for: 1.) aten.mean.dim 2.) aten.var.dim 3.) aten.sum.dim_IntList op Also, it fixes the lowering of `aten.mean.dim` and `aten.sum.dim_IntList` for handling the cases of empty dim list. Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com |
||
|---|---|---|
| .. | ||
| TorchToLinalg | ||
| TorchToMhlo | ||
| TorchToSCF | ||
| TorchToStd | ||
| TorchToTMTensor | ||
| TorchToTosa | ||
| Utils | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||