NiccoloN
33 lines
1.3 KiB
C++
33 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include "mlir/Dialect/Arith/IR/Arith.h"
|
|
#include "mlir/IR/PatternMatch.h"
|
|
#include "mlir/IR/Value.h"
|
|
#include "mlir/Transforms/FoldUtils.h"
|
|
|
|
namespace onnx_mlir {
|
|
|
|
mlir::Block* getHostConstantBlock(mlir::Operation* anchorOp);
|
|
|
|
mlir::Value getOrCreateHostConstant(mlir::Operation* anchorOp,
|
|
mlir::Attribute value,
|
|
mlir::Type type,
|
|
mlir::OperationFolder& folder);
|
|
|
|
mlir::Value getOrCreateHostConstant(mlir::Operation* anchorOp,
|
|
mlir::Attribute value,
|
|
mlir::Type type,
|
|
mlir::RewriterBase& rewriter);
|
|
|
|
mlir::Value getOrCreateHostConstantLike(mlir::arith::ConstantOp constantOp, mlir::OperationFolder& folder);
|
|
|
|
mlir::Value getOrCreateHostIndexConstant(mlir::Operation* anchorOp, int64_t value, mlir::OperationFolder& folder);
|
|
|
|
mlir::Value getOrCreateHostIndexConstant(mlir::Operation* anchorOp, int64_t value, mlir::RewriterBase& rewriter);
|
|
|
|
mlir::Value getOrCreateHostI32Constant(mlir::Operation* anchorOp, int32_t value, mlir::OperationFolder& folder);
|
|
|
|
mlir::Value getOrCreateHostI64Constant(mlir::Operation* anchorOp, int64_t value, mlir::OperationFolder& folder);
|
|
|
|
} // namespace onnx_mlir
|