llvm-project/mlir/python
MaheshRavishankar 0b33890f45 [mlir][Linalg] Add ConvolutionOpInterface.
Add an interface that allows grouping together all covolution and
pooling ops within Linalg named ops. The interface currently
- the indexing map used for input/image access is valid
- the filter and output are accessed using projected permutations
- that all loops are charecterizable as one iterating over
  - batch dimension,
  - output image dimensions,
  - filter convolved dimensions,
  - output channel dimensions,
  - input channel dimensions,
  - depth multiplier (for depthwise convolutions)

Differential Revision: https://reviews.llvm.org/D109793
2021-09-20 10:41:10 -07:00
..
mlir [mlir][Linalg] Add ConvolutionOpInterface. 2021-09-20 10:41:10 -07:00
.style.yapf
CMakeLists.txt [mlir][python] Simplify python extension loading. 2021-09-03 00:43:28 +00:00
requirements.txt Exclude pybind11 2.7.0 from MLIR python requirements. 2021-07-20 06:56:05 -07:00