mirror of https://github.com/llvm/torch-mlir
809f240f01
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. |
||
|---|---|---|
| .. | ||
| test | ||
| torch_mlir | ||
| torch_mlir_e2e_test | ||
| CMakeLists.txt | ||
| TorchMLIRModule.cpp | ||