torch-mlir/lib
Henry Tu 0c35e607b3 Add static shape for scalar tensors (#833)
* Assume zero rank tensors are scalar

* Run RefineTypes pass on JIT Graph

* Rollback assumption that zero rank tensors are scalar

* Set numSizes to -1 for non-ranked tensors

* Rename RefineTypes to RefineTupleTypes
2022-07-30 09:40:02 -04:00
..
CAPI Add static shape for scalar tensors (#833) 2022-07-30 09:40:02 -04:00
Conversion [MLIR][TORCH} Fix empty dim cases for the .dim ops 2022-07-29 11:08:57 +05:30
Dialect [MLIR][TORCH] Add decomposition for aten.std.dim Op 2022-07-29 23:52:54 +05:30
RefBackend llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07: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