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" ``` |
||
---|---|---|
.. | ||
jit_ir_common | ||
ltc | ||
onnx_c_importer | ||
pt1 | ||
CMakeLists.txt |