torch-mlir/lib/Dialect/Torch
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
..
IR Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00
Transforms Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
CMakeLists.txt Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00