llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/python
History
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
..
torch_mlir Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
torch_mlir_torchscript Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
torch_mlir_torchscript_e2e_test_configs Add E2E support for tests with heavy dependencies (heavydep tests). 2021-08-03 14:09:56 -07:00
torch_mlir_utils/codegen Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00
CMakeLists.txt Rework the python build to a static assembly of MLIR+NPCOMP (#251) 2021-07-27 16:10:10 -07:00