Commit 5bc59ce

authored

[TorchToLinalg] Support lowering MaxPool3dWithIndices (llvm#3652)

Support torch.MaxPool3dWithIndices lowering to linalg backend.

1 parent b92e618 commit 5bc59ceCopy full SHA for 5bc59ce

5 files changed

+506

-217

lines changed

lib
- Conversion/TorchToLinalg
  - Pooling.cpp
- Dialect/Torch/Transforms
  - AbstractInterpLibrary.cpp
projects/pt1
- e2e_testing
  - xfail_sets.py
- python
  - torch_mlir/jit_ir_importer/build_tools
    - abstract_interp_lib_gen.py
  - torch_mlir_e2e_test/test_suite
    - pooling.py

+506

-217

lines changed

Comments

(0)