torch-mlir/include/torch-mlir-c
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
..
Dialects.h bump llvm tag to 3580daa (#1078) 2022-07-18 16:49:03 -07:00
Registration.h Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00
TorchOps.h Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
TorchTypes.h Add static shape for scalar tensors (#833) 2022-07-30 09:40:02 -04:00