fix convert value tuple

pull/915/head
zhaozhenlong 5 years ago
parent 17de1b7890
commit de6ca6bd2f

@ -36,7 +36,6 @@ GeTensor ConvertAnyUtil(const ValuePtr &value, const AnyTraits<mindspore::tensor
std::vector<int64_t> ConvertAnyUtil(const ValuePtr &value, const std::string &name,
const AnyTraits<std::vector<int64_t>>) {
MS_EXCEPTION_IF_NULL(value);
int64_t data = GetValue<int>(value);
std::vector<int64_t> list;
if (name == "pad") {
if (!value->isa<ValueSequeue>()) {
@ -49,6 +48,7 @@ std::vector<int64_t> ConvertAnyUtil(const ValuePtr &value, const std::string &na
(void)std::transform(vec->value().begin(), vec->value().end(), list.begin()+2,
[](const ValuePtr &val) { return static_cast<int64_t>(GetValue<int>(val)); });
} else {
int64_t data = GetValue<int>(value);
int size = 2; // 2 int in list
list = TransformUtil::ConvertIntToList(data, size);
}

Loading…
Cancel
Save