Method updateRange

tools/quantization/TensorStatistic.cpp:52–79

    }
}
void TensorStatistic::updateRange() {
    if (mUpdatedRangeFlags) {
        return;
    }
    mUpdatedRangeFlags = true;
    auto tmpTensor = mOriginTensor;
    bool res = mOriginTensor->copyToHostTensor(mHostTensor.get());
    if (res) {
        tmpTensor = mHostTensor.get();
    }
    int size = tmpTensor->elementSize();
    float* dataPtr = tmpTensor->host<float>();
    auto minValue = mRange.first;
    auto maxValue = mRange.second;
    for (int i = 0; i < size; ++i) {
        minValue = std::min(minValue, dataPtr[i]);
        maxValue = std::max(maxValue, dataPtr[i]);
    }
    mRange.first = minValue;
    mRange.second = maxValue;
//    if (mRange.first > 0.0f) {
//        mRange.first = 0.0f;
//    }
//    if (mRange.second < 0.0f) {
//        mRange.second = 0.0f;
//    }
    mVisited = true;
}

void TensorStatistic::resetDistribution() {
    auto maxValue         = std::max(fabsf(mRange.second), fabsf(mRange.first));