Method ImportOutputTensor

src/armnn/LoadedNetwork.cpp:1330–1358

Note: We can only import the output pointer if all of the following hold true: a) The imported pointer is aligned sufficiently b) The tensor has zero padding c) There is only one connection to the OutputSlot and it is to an OutputLayer. d) The output pointer is allocated via malloc. (Other types will be supported in a later release) e) m_IsExportEnabled must be set to true

// d) The output pointer is allocated via malloc. (Other types will be supported in a later release)
// e) m_IsExportEnabled must be set to true
void LoadedNetwork::ImportOutputTensor(const Tensor& outputTensor, ITensorHandle* outputTensorHandle)
{
    if (!outputTensorHandle)
    {
        throw armnn::NullPointerException("Data should have been allocated.");
    }

    MemorySourceFlags importFlags = outputTensorHandle->GetImportFlags();
    if (CheckFlag(importFlags, m_NetworkProperties.m_OutputSource))
    {
        std::unique_ptr<ITensorHandle> tensorHandle =
                std::make_unique<PassthroughTensorHandle>(outputTensor.GetInfo(),
                                                          outputTensor.GetMemoryArea());

        void* mem = tensorHandle->Map(false);
        bool importOk = outputTensorHandle->Import(mem, m_NetworkProperties.m_OutputSource);
        tensorHandle->Unmap();

        if (!importOk)
        {
            throw MemoryExportException("ImportOutputTensor: Memory Export failed");
        }
    }
    else
    {
        throw MemoryExportException("ImportOutputTensor: Memory Export failed, attempting to export Input Layer");
    }

}

void CopyToOutputTensor(const Tensor& outputTensor, ITensorHandle* outputTensorHandle)
{