#pragma once

#include "llvm/ADT/ArrayRef.h"

#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/PatternMatch.h"

namespace onnx_mlir {

mlir::Value materializePaddedContractionInput(
    mlir::Value input,
    mlir::RankedTensorType paddedType,
    mlir::PatternRewriter& rewriter,
    mlir::Location loc);

mlir::FailureOr<mlir::Value> materializeTransposedContractionConstant(
    mlir::Value input,
    mlir::RankedTensorType resultType,
    llvm::ArrayRef<int64_t> permutation,
    mlir::PatternRewriter& rewriter,
    mlir::Location loc);

} // namespace onnx_mlir