torch-mlir/test/Dialect/ATen
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
..
aten_addmm.mlir Add ATen Dialect (#16) 2020-08-12 19:28:04 -07:00
aten_as_strided.mlir Add ATen Dialect (#16) 2020-08-12 19:28:04 -07:00
aten_batchnorm.mlir Add ATen Dialect (#16) 2020-08-12 19:28:04 -07:00
aten_maxpool2d.mlir Add ATen Dialect (#16) 2020-08-12 19:28:04 -07:00
recognize_aten_kernels.mlir Import ATen conv2d conversion and test (#180) 2021-03-12 17:21:16 -08:00