ilgeco
36 lines
1.3 KiB
C++
36 lines
1.3 KiB
C++
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/Common/Common.hpp"
|
|
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/Patterns.hpp"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace onnx_mlir {
|
|
|
|
void populatePrePatterns(RewritePatternSet& patterns, MLIRContext* ctx) { populateGeneratedPrePatterns(patterns, ctx); }
|
|
|
|
void populateConversionPatterns(RewritePatternSet& patterns, MLIRContext* ctx) {
|
|
populateGeneratedConversionPatterns(patterns, ctx);
|
|
populateElementwisePatterns(patterns, ctx);
|
|
populateMatMulRewritePatterns(patterns, ctx);
|
|
populateGemmPatterns(patterns, ctx);
|
|
populateConvPatterns(patterns, ctx);
|
|
populatePoolPatterns(patterns, ctx);
|
|
populateReduceMeanPatterns(patterns, ctx);
|
|
populateReluPatterns(patterns, ctx);
|
|
populateSigmoidPatterns(patterns, ctx);
|
|
populateSoftmaxPatterns(patterns, ctx);
|
|
populateConcatPatterns(patterns, ctx);
|
|
populateFlattenPatterns(patterns, ctx);
|
|
populateGatherPatterns(patterns, ctx);
|
|
populateResizePatterns(patterns, ctx);
|
|
populateReshapePatterns(patterns, ctx);
|
|
populateSlicePatterns(patterns, ctx);
|
|
populateSplitPatterns(patterns, ctx);
|
|
populateTransposePatterns(patterns, ctx);
|
|
}
|
|
|
|
void populatePostPatterns(RewritePatternSet& patterns, MLIRContext* ctx) {
|
|
populateWeightPromotionPatterns(patterns, ctx);
|
|
}
|
|
|
|
} // namespace onnx_mlir
|