#pragma once

#include "mlir/IR/BuiltinAttributes.h"

#include "llvm/ADT/SmallVector.h"

#include <cstddef>
#include <optional>

namespace onnx_mlir {

int64_t getI64Attr(mlir::ArrayAttr attr, size_t index);

int64_t getOptionalI64Attr(std::optional<mlir::ArrayAttr> attr, size_t index, int64_t defaultValue);

llvm::SmallVector<int64_t> getI64ArrayAttrValues(mlir::ArrayAttr attr);

} // namespace onnx_mlir