mirror of https://github.com/llvm/torch-mlir
df02692726
Added support for dynamic shapes in `flattenusingints` op in tosa dialect. Due to this some Argmax tests pass This PR fixes this issue https://github.com/llvm/torch-mlir/issues/3004 The following tests pass after this PR ``` 1. "ArgmaxIntModule_basic" 2. "ArgmaxIntModule_multiple_maxs" 3. "ArgmaxModule_basic" ``` |
||
---|---|---|
.. | ||
CMakeLists.txt | ||
TorchToTosa.cpp | ||
TosaLegalizeCommon.cpp | ||
TosaLegalizeUtils.cpp |