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. |
||
|---|---|---|
| .. | ||
| basic.mlir | ||
| elementwise.mlir | ||
| flatten.mlir | ||
| pooling.mlir | ||
| unsqueeze.mlir | ||