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. |
||
---|---|---|
.. | ||
ATen | ||
Basicpy | ||
Numpy | ||
Refback | ||
Refbackrt | ||
TCF | ||
TCP | ||
Torch |