ilgeco
19 lines
409 B
C++
19 lines
409 B
C++
#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
|