| 84 | } |
| 85 | |
| 86 | int ScaleCalculator::SetQuantizeMethod(CalibrationMethod method) { |
| 87 | if (method != MIN_MAX && method != KL_DIVERGENCE) { |
| 88 | LOGE("invalid method (%d) for blob quantization!\n", method); |
| 89 | return -1; |
| 90 | } |
| 91 | |
| 92 | cali_method_ = method; |
| 93 | return 0; |
| 94 | } |
| 95 | |
| 96 | void ScaleCalculator::SetMergeChannel(bool merge) { |
| 97 | merge_channel_ = merge; |