torch-mlir/lib
Ramiro Leal-Cavazos 298eeb79ca
[LINALG] Add handling of unknown dimension in size list of `view` op (#633)
The view op allows for the new shape argument to have a -1 value for
one of the dimensions, and the op is expected to deduce the size of
that dimension by looking at the sizes of the other dimensions and
comparing it to the total number of elements in the original
tensor. This commit adds this functionality.
2022-03-02 13:35:01 -08:00
..
CAPI [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
Conversion [LINALG] Add handling of unknown dimension in size list of `view` op (#633) 2022-03-02 13:35:01 -08:00
Dialect Add aten.hardtanh e2e support. 2022-03-02 12:28:06 -05:00
RefBackend LLVM bump 2022-02-16 15:28:13 -05: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