#pragma once

#include "mlir/Dialect/MemRef/IR/MemRef.h"

namespace onnx_mlir {
namespace pim {

mlir::IntegerAttr getMemRefSizeInBytesAttr(mlir::OpBuilder& builder, mlir::Value memref);

} // namespace pim
} // namespace onnx_mlir
