experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Unsigned int cannott be < 0

Browse Source
This commit is contained in:
Gaurav Singh 2020-01-12 08:34:00 -05:00
parent bc15deffb8
commit b693308b33
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/c/c_api.cc
View File

@ -1344,6 +1344,10 @@ void TF_OperationGetAttrString(TF_Operation* oper, const char* attr_name,
InvalidArgument("Attribute '", attr_name, "' is not a string"); InvalidArgument("Attribute '", attr_name, "' is not a string");
return; return;
} }
if (max_length == 0) {
InvalidArgument("Attribute '", max_length, "' is zero");
return;
}
const auto& s = attr->s(); const auto& s = attr->s();
std::memcpy(value, s.data(), std::min<size_t>(s.length(), max_length)); std::memcpy(value, s.data(), std::min<size_t>(s.length(), max_length));
} }