llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion
History
Vivek Khandelwal f15d257aac [MLIR][TORCH] Add support for ceil_mode = true for pooling ops 
This commit adds support for aten.max_pool2d, aten.max_pool2d_with_indices,
and aten.avg_pool2d op for the cases where ceil_mode = true.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2022-05-11 12:52:47 +05:30
..
TorchToLinalg [MLIR][TORCH] Add support for ceil_mode = true for pooling ops 2022-05-11 12:52:47 +05:30
TorchToSCF [TORCH][MLIR]Add lowering for control flow operations. 2022-04-29 16:25:58 +05:30
TorchToStd [MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op 2022-05-05 21:48:35 +05:30
TorchToTMTensor llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
TorchToTosa llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Utils llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30