提交 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, ...@@ -255,12 +255,14 @@ void Executor<Device, T>::LoadMemory(void **data,
case VARTYPE_TYPE_INT8: case VARTYPE_TYPE_INT8:
LoadMemInternal<int8_t>( LoadMemInternal<int8_t>(
reinterpret_cast<void **>(data_buf), 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; break;
case VARTYPE_TYPE_INT32: case VARTYPE_TYPE_INT32:
LoadMemInternal<int>(reinterpret_cast<void **>(data_buf), LoadMemInternal<int>(
reinterpret_cast<void *>(tensor->mutable_data<T>()), reinterpret_cast<void **>(data_buf),
tensor->numel()); reinterpret_cast<void *>(tensor->mutable_data<int>()),
tensor->numel());
break; break;
default: default:
LOG(kLOG_ERROR) << "data type is not supported"; 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.
先完成此消息的编辑!
想要评论请 注册