That mix of dialects (used by linalg-on-tensors backends) are pretty orthogonal, so what dialects are in the output depend on what the program is doing (is it adding two numbers? is it adding two tensors? does it have an if statement, etc.).
What dialects are you wanting Torch-MLIR to output? Also, we can provide you IR that is only in the torch dialect if that is easier.
Thanks for your answer! No I was looking at the complex example provided here. The MLIR generated from this example uses multiple dialects (namely tensor, arith, standard and linalg).
Yes that would be great! I did try the various passes on the example RESNET18 code provided, but it always produces the dialect mix. I was looking for a conversion to a single dialect (such as you mentioned tensor) or dialects which are much more common in the MLIR community (something like arith, standard and affine).