forked from OSchip/llvm-project
[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:
parent
4ede396849
commit
689096965d
|
@ -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
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue