torch-mlir/frontends/pytorch/e2e_testing/torchscript
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.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
batchnorm.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
conv.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
elementwise.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
main.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
mlp.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
quantized_models.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
vision_models.py Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
xfail_sets.py Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00