#pragma once

#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/Support/LogicalResult.h"

#include "llvm/ADT/SmallVector.h"

#include <optional>

namespace onnx_mlir {

int64_t normalizeAxis(int64_t axis, int64_t rank);

mlir::FailureOr<int64_t> normalizeAxisChecked(int64_t axis, int64_t rank);

int64_t normalizeIndex(int64_t index, int64_t dimSize);

mlir::FailureOr<llvm::SmallVector<int64_t>> normalizeAxesChecked(std::optional<mlir::ArrayAttr> axesAttr, int64_t rank);

} // namespace onnx_mlir