Add preprocessing TransposeExtractConcat pass

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

http://iree.dev/

Apache License 2.0

2.58k stars 577 forks source link

Closed IanWood1 closed 2 months ago

IanWood1 commented 3 months ago

IanWood1 commented 3 months ago

Talked with Mahesh yesterday about this. Notes:

Must check that indexing maps are identity (like when there is a affine constant 0)
gridsampler op from torch-mlir should be using linalg.index

AmosLewis commented 2 months ago