torch-mlir/lib
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
..
CAPI Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Conversion [onnx] Adding lowering for `onnx.Size` operation (#2985) 2024-03-06 17:01:05 -08:00
Dialect [onnx] Migrate `onnx.ReduceMax` to match `onnx.ReduceMin` (#2981) 2024-03-06 16:48:21 -08:00
RefBackend [TorchToLinalg] Lower aten.cat to tensor.concat (#2650) 2023-12-15 15:45:32 -05:00
CMakeLists.txt [Stablehlo] add refbackend (#2712) 2024-02-16 01:08:48 +08:00
InitAll.cpp [Stablehlo] add refbackend (#2712) 2024-02-16 01:08:48 +08:00