You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Added support for dynamic shapes in `flattenusingints` op in tosa
dialect. Due to this some Argmax tests pass
This PR fixes this issue #3004
The following tests pass after this PR
```
1. "ArgmaxIntModule_basic"
2. "ArgmaxIntModule_multiple_maxs"
3. "ArgmaxModule_basic"
```
flattenusingints
op currently only supports static shapes.torch-mlir/lib/Conversion/TorchToTosa/TorchToTosa.cpp
Lines 2488 to 2491 in a3fe130
Due to this tests like argmax fail. Sending a PR for fix
The text was updated successfully, but these errors were encountered: