mirror of https://github.com/llvm/torch-mlir
306ed62edd
There were two issues related to `ignore_index` being set (1) the onnx-to-linalg pass as not reading the value correctly (2) the mean pass was not considering the `ignore_index` value For (2) when taking the mean we need to know how many of the values were considered in the sum and therefore we cannot divide by the total number of elements. Adding a summation across the total number should correct this issue. |
||
|---|---|---|
| .. | ||
| TorchConversionToMLProgram | ||
| TorchOnnxToTorch | ||
| TorchToArith | ||
| TorchToLinalg | ||
| TorchToSCF | ||
| TorchToStablehlo | ||
| TorchToTMTensor | ||
| TorchToTensor | ||
| TorchToTosa | ||
| Utils | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||