torch-mlir icon indicating copy to clipboard operation
torch-mlir copied to clipboard
verified publisher icon llvm

Dynamic shape support needed for flatten using ints in tosa dialect

Open Abhishek-TyRnT opened this issue 1 year ago • 0 comments

flattenusingints op currently only supports static shapes. https://github.com/llvm/torch-mlir/blob/a3fe130f73f160c4d5b984f8de5b26aa471cfc9a/lib/Conversion/TorchToTosa/TorchToTosa.cpp#L2488-L2491

Due to this tests like argmax fail. Sending a PR for fix

Abhishek-TyRnT avatar Mar 10 '24 14:03 Abhishek-TyRnT