torch-mlir/frontends/pytorch/e2e_testing/torchscript
Yi Zhang 89d4931324 Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d
1. Add m_TorchConstantIntList
2. Lowering for aten.conv2d
3. Lowering aten.AdaptiveAvgPool2d
2021-07-09 15:04:29 -07:00
..
basic.py Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00
batchnorm.py Add torch.aten.batch_norm Linalg lowering support 2021-06-22 16:45:21 -07:00
conv.py Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00
elementwise.py Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00
main.py Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00
mlp.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
quantized_models.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
vision_models.py Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
xfail_sets.py Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00