torch-mlir/lib/Conversion/TorchOnnxToTorch
Rob Suderman c15f1a2bd2
[onnx] Adding lowering for `onnx.Size` operation (#2985)
We can support `onnx.Size` by requesing the size of each dimensions and
taking the product of the results, then packing it into a tensor.

---------

Co-authored-by: Scott Todd <scott.todd0@gmail.com>
2024-03-06 17:01:05 -08:00
..
CMakeLists.txt [MLIR][ONNX] Add OnnxToTorch support for Maxpool Op (#2695) 2024-01-12 14:54:38 -08:00
DefaultDomainAtoF.cpp [onnx] Add support for constants of `i1`s (#2978) 2024-03-05 13:55:13 -08:00
DefaultDomainGtoP.cpp [onnx][torch] Gridsampler E2E test and corrections of gridsampler (#2987) 2024-03-06 10:56:58 -08:00
DefaultDomainQtoZ.cpp [onnx] Adding lowering for `onnx.Size` operation (#2985) 2024-03-06 17:01:05 -08:00
PassDetail.h Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Passes.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Patterns.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
TorchOnnxToTorch.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Utils.cpp [onnx] Convert `onnx.QLinearConv` to `torch` (#2851) 2024-02-05 16:09:41 -08:00