torch-mlir/lib/Dialect/Numpy/IR
Stella Laurenzo 6a50efd046 Extend the CPA type inference to work on numpy types/ops.
* Adds an op interface for adding CPA constraints.
* Adds a type conversion hook for handling built-in types (that we can't have adopt our interface).
* Converts tensor<> to object(!Tensor, [e:<type>]) just like NdArray.
* Implement a few numpy ops far enough to do dtype inference for simple sequences.
2020-07-03 18:16:34 -07:00
..
CMakeLists.txt Add partial evaluator for explicit numpy ufuncs. 2020-06-29 15:27:39 -07:00
NumpyDialect.cpp Add NdArray type inference conversion. 2020-07-03 16:38:10 -07:00
NumpyOps.cpp Extend the CPA type inference to work on numpy types/ops. 2020-07-03 18:16:34 -07:00