saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Return a Status instead of throw an exception in GetAttrs (#10534)

Browse source
This commit is contained in:
Changming Sun 2022-02-14 13:24:35 -08:00 committed by GitHub
parent 3f37609994
commit 270dec7327
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
onnxruntime/core/framework/op_node_proto_helper.cc
View file

@ -90,34 +90,35 @@ inline constexpr int ArrayTypeToAttributeType<std::string>() {
} \
}
#define ORT_DEFINE_GET_ATTRS(IMPL_T, T, list) \
template <> \
template <> \
Status OpNodeProtoHelper<IMPL_T>::GetAttrs<T>( \
const std::string& name, std::vector<T>& values) const { \
const AttributeProto* attr = TryGetAttribute(name); \
if (!attr) { \
return Status(ONNXRUNTIME, FAIL, "No attribute with this name is defined."); \
} \
values.reserve(attr->list##_size()); \
for (int i = 0; i < attr->list##_size(); ++i) { \
values.push_back(static_cast<T>(attr->list(i))); \
} \
return Status::OK(); \
} \
template <> \
template <> \
Status OpNodeProtoHelper<IMPL_T>::GetAttrs<T>( \
const std::string& name, gsl::span<T> values) const { \
const AttributeProto* attr = TryGetAttribute(name); \
if (!attr) { \
return Status(ONNXRUNTIME, FAIL, "No attribute with this name is defined."); \
} \
ORT_ENFORCE(values.size() == static_cast<size_t>(attr->list##_size())); \
for (int i = 0; i < attr->list##_size(); ++i) { \
values[i] = static_cast<T>(attr->list(i)); \
} \
return Status::OK(); \
#define ORT_DEFINE_GET_ATTRS(IMPL_T, T, list) \
template <> \
template <> \
Status OpNodeProtoHelper<IMPL_T>::GetAttrs<T>( \
const std::string& name, std::vector<T>& values) const { \
const AttributeProto* attr = TryGetAttribute(name); \
if (!attr) { \
return Status(ONNXRUNTIME, FAIL, "No attribute with this name is defined."); \
} \
values.reserve(attr->list##_size()); \
for (int i = 0; i < attr->list##_size(); ++i) { \
values.push_back(static_cast<T>(attr->list(i))); \
} \
return Status::OK(); \
} \
template <> \
template <> \
Status OpNodeProtoHelper<IMPL_T>::GetAttrs<T>( \
const std::string& name, gsl::span<T> values) const { \
const AttributeProto* attr = TryGetAttribute(name); \
if (!attr) { \
return Status(ONNXRUNTIME, FAIL, "No attribute with this name is defined."); \
} \
ORT_RETURN_IF(values.size() != static_cast<size_t>(attr->list##_size()), \
"GetAttrs failed. Expect values.size()=" , (attr->list##_size()) , ", got " , values.size()); \
for (int i = 0; i < attr->list##_size(); ++i) { \
values[i] = static_cast<T>(attr->list(i)); \
} \
return Status::OK(); \
}
// Will not work for std::strings