mirror of https://github.com/llvm/torch-mlir
297c270980
The old lowering only had logic for 2d (i.e. images). this patch allows interpolation for n spatial dims, which is required for some 3d vision models such as - onnx/models/pytorch-3dunet_vaiq_int8 which successfully compiles and runs with this patch. |
||
|---|---|---|
| .. | ||
| basic.mlir | ||
| broadcast.mlir | ||
| convolution.mlir | ||
| elementwise.mlir | ||
| flatten.mlir | ||
| gridsampler.mlir | ||
| pooling.mlir | ||
| resize.mlir | ||
| sparse.mlir | ||
| unsqueeze.mlir | ||
| view.mlir | ||
| view_strict.mlir | ||