mirror of https://github.com/llvm/torch-mlir
ec6d06aa86
- aten::relu_, aten::max_pool2d, aten::adaptive_avg_pool2d, aten::batch_norm, aten::conv2d No aten-to-linalg conversion for the latter ones, as they are fairly substantial. At this point, I'm trying to get shape inference and stuff working for them and the IR cleaned up. |
||
|---|---|---|
| .. | ||
| torch_mlir | ||
| torch_mlir_utils/codegen | ||
| CMakeLists.txt | ||