NiccoloN
d09e76c8f9
Validate Operations / validate-operations (push) Has been cancelled
fix reshape lowering add support for grouped-convolution lowering quieter verifier with capped error messages
24 lines
720 B
C++
24 lines
720 B
C++
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/Common/Common.hpp"
|
|
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/ConversionPatterns.hpp"
|
|
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/PrePatterns.hpp"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace onnx_mlir {
|
|
|
|
namespace {
|
|
|
|
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/ONNXToSpatial.hpp.inc"
|
|
|
|
} // namespace
|
|
|
|
void populatePrePatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx) {
|
|
patterns.add<onnxToArithConstant>(ctx);
|
|
patterns.add<convAddToConvWithBiasLeft>(ctx);
|
|
patterns.add<convAddToConvWithBiasRight>(ctx);
|
|
patterns.add<matMulAddToGemm>(ctx);
|
|
patterns.add<removeFlattenSameShape>(ctx);
|
|
}
|
|
|
|
} // namespace onnx_mlir
|