NiccoloN
21 lines
505 B
C++
21 lines
505 B
C++
#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
|