提交 a77f09c2 编写于 作者: Z zhaojiaying01

[Mobile][ARM]Fix bug to support PaddleSlim quantize trained model. test=mobile

上级 7950edd7
......@@ -255,12 +255,14 @@ void Executor<Device, T>::LoadMemory(void **data,
case VARTYPE_TYPE_INT8:
LoadMemInternal<int8_t>(
reinterpret_cast<void **>(data_buf),
reinterpret_cast<void *>(tensor->mutable_data<T>()), tensor->numel());
reinterpret_cast<void *>(tensor->mutable_data<int8_t>()),
tensor->numel());
break;
case VARTYPE_TYPE_INT32:
LoadMemInternal<int>(reinterpret_cast<void **>(data_buf),
reinterpret_cast<void *>(tensor->mutable_data<T>()),
tensor->numel());
LoadMemInternal<int>(
reinterpret_cast<void **>(data_buf),
reinterpret_cast<void *>(tensor->mutable_data<int>()),
tensor->numel());
break;
default:
LOG(kLOG_ERROR) << "data type is not supported";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册