torch-mlir/test/Conversion/TorchToLinalg
Yi Zhang 0342b73bf1 Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering
- torch.aten.flatten.using_ints to linalg lowering
- torch.aten.max_pool2d to linalg lowering
- Support torch.aten.conv2d for more flexible dilation and strides values
2021-08-04 12:00:43 -04:00
..
basic.mlir Bump llvm-project to da289a174fc6617c7be37be2947480510fd4f02a 2021-07-07 13:57:29 -07:00
elementwise.mlir Bump llvm-project to da289a174fc6617c7be37be2947480510fd4f02a 2021-07-07 13:57:29 -07:00
flatten.mlir Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
unsqueeze.mlir Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00