torch-mlir/include/npcomp/Dialect/Torch/IR
Yi Zhang 89d4931324 Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d
1. Add m_TorchConstantIntList
2. Lowering for aten.conv2d
3. Lowering aten.AdaptiveAvgPool2d
2021-07-09 15:04:29 -07:00
..
.gitignore Significantly restructure torch/aten import design. 2021-05-19 13:37:39 -07:00
CMakeLists.txt Bump llvm-project to da289a174fc6617c7be37be2947480510fd4f02a 2021-07-07 13:57:29 -07:00
GeneratedAtenOps.td Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00
GeneratedPrimOps.td Add CastOpInterface to torch.prim.unchecked_cast. 2021-06-23 08:07:45 -07:00
GeneratedQuantizedOps.td Add a !torch.float type. 2021-06-17 09:24:18 -07:00
TorchBase.td Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
TorchDialect.h Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
TorchOps.h Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d 2021-07-09 15:04:29 -07:00
TorchOps.td Make MaximizeValueSemantics a bit smarter. 2021-06-22 16:48:57 -07:00
TorchTraits.h Significantly restructure torch/aten import design. 2021-05-19 13:37:39 -07:00
TorchTypes.h Move Torch type implementation code into TorchTypes.cpp 2021-06-10 16:46:47 -07:00
TorchTypes.td Add a !torch.float type. 2021-06-17 09:24:18 -07:00