ilgeco
24 lines
572 B
C++
24 lines
572 B
C++
#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
|