torch-mlir/lib/Dialect/Torch/Utils
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
..
CMakeLists.txt Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
TorchUpstream.cpp Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
Utils.cpp importer: add initial support for loading Float16 tensors (#1169) 2022-08-08 12:37:31 +08:00