llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/CAPI
History
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
..
CMakeLists.txt Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00
Dialects.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Registration.cpp Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00
TorchOps.cpp Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
TorchTypes.cpp Add static shape for scalar tensors (#833) 2022-07-30 09:40:02 -04:00