Method updateDistribution

tools/quantization/TensorStatistic.cpp:92–114

    // fabsf(mRangePerChannel[0].first)));
}
void TensorStatistic::updateDistribution() {
    if (mUpdatedDistributionFlag) {
        return;
    }
    mUpdatedDistributionFlag = true;
    auto tmpTensor = mOriginTensor;
    bool res = mOriginTensor->copyToHostTensor(mHostTensor.get());
    if (res) {
        tmpTensor = mHostTensor.get();
    }
//    float midValue = (mRange.second + mRange.first) / 2.0f;
    float midValue = 0.f;
    auto ptr = tmpTensor->host<float>();
    for (int i = 0; i < tmpTensor->elementSize(); ++i) {
        auto data = ptr[i] - midValue;
        if (abs(data) <= 1e-6) {
            continue;
        }
        int index = static_cast<int>(fabs(data) * mInterval);
        index = std::min(index, mBinNumber - 1);
        mDistribution[index] += 1.0f;
    }
}

void TensorStatistic::setThresholdMethod(GET_THRESHOLD_METHOD thresholdMethod) {
    mThresholdMethod = thresholdMethod;