[mlir][Linalg] Conv ops lowering to std calls added.

Lowering of newly defined Conv ops in TC syntax to standard
dialect is not supported and therefore this commit adds support
for it.

Differential Revision: https://reviews.llvm.org/D84840
This commit is contained in:
Jakub Lichman 2020-08-03 07:57:06 +00:00
parent 4ede396849
commit 689096965d
1 changed files with 10 additions and 1 deletions

View File

@ -244,7 +244,16 @@ void mlir::populateLinalgToStandardConversionPatterns(
LinalgOpConversion<DotOp>,
LinalgOpConversion<BatchMatmulOp>,
LinalgOpConversion<MatvecOp>,
LinalgOpConversion<MatmulOp>>(ctx);
LinalgOpConversion<MatmulOp>,
LinalgOpConversion<ConvWOp>,
LinalgOpConversion<ConvNWCOp>,
LinalgOpConversion<ConvNCWOp>,
LinalgOpConversion<ConvHWOp>,
LinalgOpConversion<ConvNHWCOp>,
LinalgOpConversion<ConvNCHWOp>,
LinalgOpConversion<ConvDHWOp>,
LinalgOpConversion<ConvNDHWCOp>,
LinalgOpConversion<ConvNCDHWOp>>(ctx);
// clang-format on
}