Function GetUIntBuffer

src/armnnTfLiteParser/TfLiteParser.cpp:298–344

}

std::vector<unsigned int> GetUIntBuffer(armnn::TensorInfo info,
                                        const TfLiteParserImpl::ModelPtr& model,
                                        size_t bufferIndex)
{
    TfLiteParserImpl::BufferRawPtr bufferPtr = TfLiteParserImpl::GetBuffer(model, bufferIndex);

    if (!bufferPtr || bufferPtr->data.empty())
    {
        throw ParseException("UInt buffer is null or empty.");
    }

    const size_t numElements = info.GetNumElements();
    std::vector<unsigned int> buffer(info.GetNumElements());
    buffer.reserve(numElements);

    if (info.GetDataType() == DataType::Signed32)
    {
        size_t requiredBytes = numElements * sizeof(int32_t);
        if (bufferPtr->data.size() < requiredBytes)
        {
            throw ParseException("UInt buffer too small for Signed32 tensor.");
        }

        ::memcpy(buffer.data(), bufferPtr->data.data(), requiredBytes);
    }
    else if (info.GetDataType() == DataType::Signed64)
    {
        size_t requiredBytes = numElements * sizeof(int64_t);
        if (bufferPtr->data.size() < requiredBytes)
        {
            throw ParseException("UInt buffer too small for Signed64 tensor.");
        }

        std::vector<uint64_t> uint64Buffer(numElements);
        ::memcpy(uint64Buffer.data(), bufferPtr->data.data(), requiredBytes);
        buffer.assign(std::begin(uint64Buffer), std::end(uint64Buffer));
    }
    else
    {
        CheckLocation location = CHECK_LOCATION();
        throw ParseException(
                fmt::format("Unsupported data type for uint buffer {}, only Signed 32 or Signed 64 are supported. {}",
                            GetDataTypeName(info.GetDataType()),
                            location.AsString()));
    }
    return buffer;
}

#define CHECK_BUFFER_SIZE(BUFFER_PTR, TENSOR_INFO, BUFFER_ID) \
    CheckBufferSize(BUFFER_PTR, TENSOR_INFO, BUFFER_ID, CHECK_LOCATION())