Method getTensorSize

source/backend/cpu/CPUBackend.cpp:682–698

}

size_t CPUBackend::getTensorSize(const Tensor* tensor, bool multiBytes) const {
    auto core = mCoreFunctions;
    size_t dataSize = 1;
    auto des = TensorUtils::getDescribe(tensor);
    for (int i = 0; i < tensor->dimensions(); i++) {
        size_t currentDimSize = tensor->length(i);
        if (des->dimensionFormat == MNN_DATA_FORMAT_NC4HW4 && 1 == i) {
            currentDimSize = UP_DIV(currentDimSize, core->pack) * core->pack;
        }
        dataSize *= currentDimSize;
    }
    if (multiBytes) {
        size_t bytes = getBytes(this, tensor);
        return dataSize * bytes;
    }
    return dataSize;
}

size_t CPUBackend::getBytes(const Backend* backend, const Tensor* output) {
    size_t bytes = output->getType().bytes();