Method usize

source/core/Tensor.cpp:409–421

}

size_t Tensor::usize() const {
    size_t dataSize = mBuffer.type.bytes();
    MNN_ASSERT(dataSize >= 1);
    auto nativeDescribe = mDescribe->mContent.get();
    for (int i = 0; i < this->buffer().dimensions; i++) {
        int currentDimSize = mBuffer.dim[i].extent;
        if (nativeDescribe->dimensionFormat == MNN_DATA_FORMAT_NC4HW4 && 1 == i) {
            currentDimSize = ALIGN_UP4(currentDimSize);
        }
        dataSize *= currentDimSize;
    }
    return dataSize;
}

int Tensor::size() const {
    return static_cast<int>(usize());