torch-mlir/lib/Conversion/TorchToTensor
Rob Suderman dd673cfa8d
[torch] Add edgecase for aten.shape_to_tensor for rank-0 input (#2962)
Currently lowering uses `tensor.from_elements` which does not allow zero
inputs. In this case we return a `tensor.empty` operation.
2024-02-28 09:47:06 -08:00
..
CMakeLists.txt [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
TorchToTensor.cpp [torch] Add edgecase for aten.shape_to_tensor for rank-0 input (#2962) 2024-02-28 09:47:06 -08:00