#pragma once

#include "mlir/IR/MLIRContext.h"
#include "mlir/Transforms/DialectConversion.h"

namespace onnx_mlir {

void populateConversionPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateConvPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateElementwisePatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateGemmPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateMatMulRewritePatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populatePoolPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateReduceMeanPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateReluPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateSigmoidPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateSoftmaxPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateConcatPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateGatherPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateResizePatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateReshapePatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populateSplitPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

} // namespace onnx_mlir