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 |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| dialects | ||
| eager_mode | ||
| __init__.py | ||
| compiler_utils.py | ||