Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_executor/
H A Dnn_executor_test.cpp153 m_outputTensorDescs.emplace_back(pair1);
154 m_outputTensorDescs.emplace_back(pair2);
493 m_outputTensorDescs.emplace_back(pair1);
494 m_outputTensorDescs.emplace_back(pair2);
524 m_outputTensorDescs.emplace_back(pair1);
525 m_outputTensorDescs.emplace_back(pair2);
561 m_outputTensorDescs.emplace_back(pair1);
562 m_outputTensorDescs.emplace_back(pair2);
735 m_outputTensorDescs.emplace_back(pair1);
736 m_outputTensorDescs.emplace_back(pair2);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnexecutor.cpp36 m_outputTensorDescs(outputTensorDescs) {} in NNExecutor()
113 if (outputIndex >= m_outputTensorDescs.size()) { in GetOutputShape()
141 return m_outputTensorDescs.size(); in GetOutputNum()
169 if (index >= m_outputTensorDescs.size()) { in CreateOutputTensorDesc()
173 if (m_outputTensorDescs[index].first == nullptr) { in CreateOutputTensorDesc()
248 if (m_outputTensorDescs.size() != outputSize) { in RunSync()
250 outputSize, m_outputTensorDescs.size()); in RunSync()
728 if (index >= m_outputTensorDescs.size()) { in SetOutput()
795 if (index >= m_outputTensorDescs.size()) { in SetOutputFromMemory()
908 if (index >= m_outputTensorDescs.size()) { in CreateOutputMemory()
[all …]
H A Dnncompiler.cpp133 m_outputTensorDescs = m_innerModel->GetOutputTensorDescs(); in NNCompiler()
144 m_outputTensorDescs.clear(); in ~NNCompiler()
557 ret = SerializeTensorsToBuffer(m_outputTensorDescs, outputTensorDescBuffer); in SaveToCacheFile()
701 m_outputTensorDescs = outputTensorDescs; in RestoreFromCacheFile()
765 if (m_outputTensorDescs.empty()) { in CreateExecutor()
771 m_backendID, m_device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs); in CreateExecutor()
H A Dnncompiler.h89 std::vector<std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType>> m_outputTensorDescs; variable
H A Dnnexecutor.h100 std::vector<std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType>> m_outputTensorDescs; variable
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp1296 m_backendID, m_device, mockIPreparedMode, m_inputTensorDescs, m_outputTensorDescs);
1328 m_backendID, m_device, mockIPreparedMode, m_inputTensorDescs, m_outputTensorDescs);
1358 m_backendID, m_device, mockIPreparedMode, m_inputTensorDescs, m_outputTensorDescs);
1388 m_backendID, m_device, mockIPreparedMode, m_inputTensorDescs, m_outputTensorDescs);
1496 m_backendID, m_device, mockIPreparedMode, m_inputTensorDescs, m_outputTensorDescs);
1853 m_outputTensorDescs.emplace_back(pair1);
1854 m_outputTensorDescs.emplace_back(pair2);
1861 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
2033 m_outputTensorDescs.emplace_back(pair1);
2034 m_outputTensorDescs.emplace_back(pair2);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_core_test/
H A Dneural_network_core_test.cpp1754 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
1792 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
1828 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
1897 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
1925 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
1961 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
1996 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
2034 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
2059 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
2084 m_backendID, device, m_preparedModel, m_inputTensorDescs, m_outputTensorDescs);
[all …]