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/python/importer
History
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
..
jit_ir importer: add initial support for loading BFloat16 tensors (#761) 2022-04-29 09:01:49 -07:00