llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/include/npcomp/Dialect/ATen/IR
History
Aaron Arthurs 4fd9b4afb5
Import ATen conv2d conversion and test (#180) 
* Import ATen conv2d conversion and test

This is a first attempt at expanding ATen-to-TCF conversion for the
conv2d operator. Eventually, this will come in use when lowering a
high-level conv-based model.
 		2021-03-12 17:21:16 -08:00
..
.gitignore Add a number of kernels and new patterns. 2020-11-04 14:36:59 -08:00
ATenDialect.h Bump llvm-project to 444822d77a7fea28aa49edf24533c987efa1b2ee 2020-12-11 14:43:38 -08:00
ATenDialect.td Bump llvm-project to bc556e5685c0f97e79fb7b3c6f15cc5062db8e36 2021-01-08 14:18:09 -08:00
ATenOpInterface.td NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00
ATenOpInterfaces.h NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00
ATenOpStatisticsUtils.h Bump llvm-project to 444822d77a7fea28aa49edf24533c987efa1b2ee 2020-12-11 14:43:38 -08:00
ATenOps.td Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
CMakeLists.txt NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00
GeneratedATenOps.cpp.inc Import ATen conv2d conversion and test (#180) 2021-03-12 17:21:16 -08:00
GeneratedATenOps.td Import ATen conv2d conversion and test (#180) 2021-03-12 17:21:16 -08:00
LegacyGeneratedATenOps.td Add aten.mm op and "test" it e2e. 2020-11-20 17:21:24 -08:00