torch-mlir

Author	SHA1	Message	Date
gpetters94	c4dcdd1e34	Add aten.flip (#817 )	2022-05-02 16:01:15 -04:00
Vivek Khandelwal	769f3a8870	[MLIR][TORCH] Add E2E support for max_pool2d_with_indices op This commit adds lowering of `max_pool2d_with_indices` op. Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>	2022-04-18 21:05:19 +05:30
gpetters94	9ec0683e92	Add 2D case for convolution (#693 )	2022-04-08 00:47:57 -04:00
Sean Silva	5d9222383c	Split up TorchToLinalg.cpp This helps keep things organized and also exposes more parallelism to the build system. It seems though that most of the compile time is actually spent in the headers though, so the wall time doesn't decrease as much as I had hoped (and now that the headers are being included multiple times, the cpu time actually increases a lot, sadly -- will try to dig into this).	2022-03-14 10:19:41 -07:00