|
|
|
@ -124,7 +124,7 @@ STATUS QuantFilter(ParamValueLitePtr weight, std::shared_ptr<PrimitiveC> primiti
|
|
|
|
|
auto dims = weight->tensor_shape();
|
|
|
|
|
if (per_channel) {
|
|
|
|
|
if (dims.size() != 4) {
|
|
|
|
|
MS_LOG(ERROR) << "weight dims size: " << dims.size() << " switch to per-layer quant mode.";
|
|
|
|
|
MS_LOG(INFO) << "weight dims size: " << dims.size() << " switch to per-layer quant mode.";
|
|
|
|
|
per_channel = false;
|
|
|
|
|
} else {
|
|
|
|
|
uint32_t channels = dims[0];
|
|
|
|
|