Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnexecutor.cpp493 m_inputTensors[index].tensor->SetBuffer(nullptr, 0); in SetInputTensorWithCurrentBuffer()
496 m_inputTensors[index].tensor = inputTensor; in SetInputTensorWithCurrentBuffer()
505 if (m_inputTensors.find(index) != m_inputTensors.end()) { in SetInputTensorWithNewBuffer()
506 if (m_inputTensors[index].isInnerMem) { in SetInputTensorWithNewBuffer()
511 m_inputTensors[index].tensor->SetBuffer(nullptr, 0); in SetInputTensorWithNewBuffer()
519 m_inputTensors[index] = exeTensor; in SetInputTensorWithNewBuffer()
602 … if ((m_inputTensors.find(index) != m_inputTensors.end()) && (m_inputTensors[index].isInnerMem)) { in SetInput()
1035 size_t inputSize = m_inputTensors.size(); in Run()
1038 inputTensors.emplace_back(m_inputTensors[i].tensor); in Run()
1082 for (auto& it : m_inputTensors) { in ~NNExecutor()
[all …]
H A Dinner_model.cpp131 m_inputTensors.clear(); in BuildFromLiteGraph()
132 …rnCode ret = ConstructNNTensorsFromLiteGraph(liteGraph, liteGraph->input_indices_, m_inputTensors); in BuildFromLiteGraph()
449 if (!m_inputTensors.empty()) { in SpecifyInputsAndOutputs()
463 std::transform(m_inputIndices.begin(), m_inputIndices.end(), std::back_inserter(m_inputTensors), in SpecifyInputsAndOutputs()
493 if (!(m_inputTensors.empty() && (m_inputIndices.empty()))) { in CheckParameters()
530 m_inputTensors.emplace_back(tensor); in SetInputsAndOutputsInfo()
710 return m_inputTensors; in GetInputTensors()
722 for (auto inputTensor : m_inputTensors) { in GetInputTensorDescs()
H A Dinner_model.h78 …std::vector<std::shared_ptr<NNTensor>> m_inputTensors; // Used to pass input tensors to compilatio… variable
H A Dnnexecutor.h111 std::unordered_map<int, ExeTensor> m_inputTensors; variable