Searched refs:m_outputTensors (Results 1 – 4 of 4) sorted by relevance
751 if (m_outputTensors.find(index) != m_outputTensors.end()) { in SetOutput()770 m_outputTensors[index] = exeTensor; in SetOutput()785 m_outputTensors[index].userBuffer = buffer; in SetOutput()787 m_outputTensors[index].isInnerMem = true; in SetOutput()815 if (m_outputTensors.find(index) != m_outputTensors.end()) { in SetOutputFromMemory()824 m_outputTensors[index] = exeTensor; in SetOutputFromMemory()836 m_outputTensors[index].isInnerMem = false; in SetOutputFromMemory()1054 if (m_outputTensors[i].isInnerMem) { in Run()1068 …status = memcpy_s(m_outputTensors[i].userBuffer, m_outputTensors[i].userBufferLength, deviceBuffer… in Run()1092 for (auto& it : m_outputTensors) { in ~NNExecutor()[all …]
138 m_outputTensors.clear(); in BuildFromLiteGraph()139 ret = ConstructNNTensorsFromLiteGraph(liteGraph, liteGraph->output_indices_, m_outputTensors); in BuildFromLiteGraph()468 …std::transform(m_outputIndices.begin(), m_outputIndices.end(), std::back_inserter(m_outputTensors), in SpecifyInputsAndOutputs()498 if (!(m_outputTensors.empty() && (m_outputIndices.empty()))) { in CheckParameters()548 m_outputTensors.emplace_back(tensor); in SetInputsAndOutputsInfo()715 return m_outputTensors; in GetOutputTensors()736 for (auto outputTensor : m_outputTensors) { in GetOutputTensorDescs()
79 …std::vector<std::shared_ptr<NNTensor>> m_outputTensors; // Used to pass output tensors to compilat… variable
112 std::unordered_map<int, ExeTensor> m_outputTensors; variable