NiccoloN
909c4acfdd
Validate Operations / validate-operations (push) Has been cancelled
remove Spatial many ops in favor of tensor ops like in pim
33 lines
1001 B
C++
33 lines
1001 B
C++
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/Common/Common.hpp"
|
|
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/ConversionPatterns.hpp"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace onnx_mlir {
|
|
|
|
namespace {
|
|
|
|
#include "src/Accelerators/PIM/Conversion/ONNXToSpatial/ONNXToSpatial.hpp.inc"
|
|
|
|
} // namespace
|
|
|
|
void populateConversionPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx) {
|
|
patterns.add<removeLRN>(ctx);
|
|
|
|
populateElementwisePatterns(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);
|
|
populateGatherPatterns(patterns, ctx);
|
|
populateResizePatterns(patterns, ctx);
|
|
populateReshapePatterns(patterns, ctx);
|
|
populateSplitPatterns(patterns, ctx);
|
|
}
|
|
|
|
} // namespace onnx_mlir
|