torch-mlir/python/torch_mlir/extras
Daniel Garvey faf7d4aaa5
[fx_importer] Add support for 0D tensors (#2870)
Adds an escape hatch from creating a DenseResourceElementsAttr for
single value tensors into DenseElementsAttr.

For 0d or 1element, splats are better as DenseElementsAttr. Don't use
DenseResourceElementsAttr for it
2024-02-06 00:19:31 -06:00
..
fx_importer.py [fx_importer] Add support for 0D tensors (#2870) 2024-02-06 00:19:31 -06:00
onnx_importer.py [onnx] Import `onnx` constants as `onnx.Constant` instead of literals (#2831) 2024-01-31 11:41:06 -08:00