torch-mlir/lib
Ashay Rane 809f240f01
importer: add initial support for loading BFloat16 tensors (#761)
This patch updates the `torch_mlir::convertTensorToMlirElementsAttr()`
method to enable the creation of tensors whose base type is BFloat16.
This patch also adds a test to validate the IR generation, and it
updates the test for importing tensors of various types.
2022-04-29 09:01:49 -07:00
..
CAPI Add minimal support for Union types. 2022-03-29 17:45:48 -07:00
Conversion [TORCH][MLIR]Add lowering for control flow operations. 2022-04-29 16:25:58 +05:30
Dialect importer: add initial support for loading BFloat16 tensors (#761) 2022-04-29 09:01:49 -07:00
RefBackend llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00
InitAll.cpp Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00