|
|
|
@ -236,7 +236,7 @@ kernel::LiteKernel *CpuConvFp32KernelCreator(const std::vector<lite::Tensor *> &
|
|
|
|
|
|
|
|
|
|
auto *weight_tensor = inputs.at(kWeightIndex);
|
|
|
|
|
auto *restore_data = weight_tensor->MutableData();
|
|
|
|
|
if (primitive->GetQuantType() == schema::QuantType_WeightQuant) {
|
|
|
|
|
if (weight_tensor->data_type() == kNumberTypeInt8 || primitive->GetQuantType() == schema::QuantType_WeightQuant) {
|
|
|
|
|
ConvolutionBaseCPUKernel::RestoreFilter(inputs.at(kWeightIndex));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -265,7 +265,7 @@ kernel::LiteKernel *CpuConvFp32KernelCreator(const std::vector<lite::Tensor *> &
|
|
|
|
|
return nullptr;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (primitive->GetQuantType() == schema::QuantType_WeightQuant) {
|
|
|
|
|
if (weight_tensor->data_type() == kNumberTypeInt8 || primitive->GetQuantType() == schema::QuantType_WeightQuant) {
|
|
|
|
|
weight_tensor->FreeData();
|
|
|
|
|
weight_tensor->SetData(restore_data);
|
|
|
|
|
}
|
|
|
|
|