llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/Conversion/TorchToLinalg
History
Tanyo Kwok 290d7755fb
importer: add initial support for loading Float16 tensors (#1169) 
follow up #761:

    This patch updates the `torch_mlir::convertTensorToMlirElementsAttr()`
    method to enable the creation of tensors whose base type is Float16.
    This patch also adds a test to validate the IR generation, and it
    updates the test for importing tensors of various types.
 		2022-08-08 12:37:31 +08:00
..
basic.mlir importer: add initial support for loading Float16 tensors (#1169) 2022-08-08 12:37:31 +08:00
elementwise.mlir linalg: lower `aten.triu` op to `linalg.generic` (#965) 2022-06-23 22:45:48 -07:00
flatten.mlir mlir: bump llvm tag to 5380e3 (#856) 2022-05-16 12:54:35 -07:00
pooling.mlir [MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op 2022-05-27 07:56:37 +05:30
unsqueeze.mlir mlir: bump llvm tag to 5380e3 (#856) 2022-05-16 12:54:35 -07:00