Home
last modified time | relevance | path

Searched refs:deviceOutputBuffer (Results 1 – 1 of 1) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnexecutor.cpp778 …void* deviceOutputBuffer = m_device->AllocateTensorBuffer(length, m_outputTensorDescs[index].first… in SetOutput() local
779 if (deviceOutputBuffer == nullptr) { in SetOutput()
784 m_outputTensors[index].tensor->SetBuffer(deviceOutputBuffer, length); in SetOutput()
918 …void* deviceOutputBuffer = m_device->AllocateTensorBuffer(length, m_outputTensorDescs[index].first… in CreateOutputMemory() local
919 if (deviceOutputBuffer == nullptr) { in CreateOutputMemory()
924 *memory = new(std::nothrow) OH_NN_Memory{deviceOutputBuffer, length}; in CreateOutputMemory()
927 m_device->ReleaseBuffer(deviceOutputBuffer); in CreateOutputMemory()
932 m_outputCreatedMem[index].emplace_back(deviceOutputBuffer); in CreateOutputMemory()