Searched refs:m_outputIndices (Results 1 – 9 of 9) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
H A D | inner_model.cpp | 461 m_outputIndices = ConstructVectorFromArray(outputIndices.data, outputIndices.size); in SpecifyInputsAndOutputs() 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() 546 m_outputIndices.emplace_back(i + inputSize); in SetInputsAndOutputsInfo() 573 if ((m_inputIndices.empty()) || (m_outputIndices.empty())) { in Build() 638 … std::transform(m_outputIndices.begin(), m_outputIndices.end(), std::back_inserter(outputIndices), in AddTensorsToLiteGraph()
|
H A D | inner_model.h | 75 std::vector<uint32_t> m_outputIndices; variable
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/neural_network_runtime_test/ |
H A D | neural_network_runtime_test.cpp | 178 m_outputIndices.data = m_outputIndexs; in InitIndices() 179 m_outputIndices.size = sizeof(m_outputIndexs) / sizeof(uint32_t); in InitIndices() 216 EXPECT_EQ(OH_NN_SUCCESS, innerModel.SpecifyInputsAndOutputs(m_inputIndices, m_outputIndices)); in SetInnerBuild() 340 m_outputIndices.data = m_outputIndexs; 341 m_outputIndices.size = sizeof(m_outputIndexs) / sizeof(uint32_t); 367 m_outputIndices.data = m_outputIndexs; 368 m_outputIndices.size = sizeof(m_outputIndexs) / sizeof(uint32_t); 516 …H_NN_ReturnCode ret = OH_NNModel_SpecifyInputsAndOutputs(model, &m_inputIndices, &m_outputIndices); 533 OH_NN_ReturnCode ret = OH_NNModel_SpecifyInputsAndOutputs(model, nullptr, &m_outputIndices); 592 EXPECT_EQ(OH_NN_SUCCESS, innerModel.SpecifyInputsAndOutputs(m_inputIndices, m_outputIndices)); [all …]
|
H A D | neural_network_runtime_test.h | 40 OH_NN_UInt32Array m_outputIndices; variable
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/ |
H A D | neural_network_runtime_test.h | 40 OH_NN_UInt32Array m_outputIndices; variable
|
H A D | neural_network_runtime_test.cpp | 179 m_outputIndices.data = m_outputIndexs; in InitIndices() 180 m_outputIndices.size = sizeof(m_outputIndexs) / sizeof(uint32_t); in InitIndices() 217 EXPECT_EQ(OH_NN_SUCCESS, innerModel.SpecifyInputsAndOutputs(m_inputIndices, m_outputIndices)); in SetInnerBuild() 393 m_outputIndices.data = m_outputIndexs; 394 m_outputIndices.size = sizeof(m_outputIndexs) / sizeof(uint32_t); 420 m_outputIndices.data = m_outputIndexs; 421 m_outputIndices.size = sizeof(m_outputIndexs) / sizeof(uint32_t); 569 …H_NN_ReturnCode ret = OH_NNModel_SpecifyInputsAndOutputs(model, &m_inputIndices, &m_outputIndices); 586 OH_NN_ReturnCode ret = OH_NNModel_SpecifyInputsAndOutputs(model, nullptr, &m_outputIndices); 645 EXPECT_EQ(OH_NN_SUCCESS, innerModel.SpecifyInputsAndOutputs(m_inputIndices, m_outputIndices)); [all …]
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_core_test/ |
H A D | neural_network_core_test.h | 40 OH_NN_UInt32Array m_outputIndices; variable
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/ |
H A D | inner_model_test.cpp | 53 std::vector<uint32_t> m_outputIndices {1}; member in NNRT::UnitTest::InnerModelTest 70 liteGraph->output_indices_ = m_outputIndices; in SetLiteGraph() 192 m_outputIndices = {6};
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/ |
H A D | inner_model_test.cpp | 53 std::vector<uint32_t> m_outputIndices {1}; member in NNRT::UnitTest::InnerModelTest 70 liteGraph->output_indices_ = m_outputIndices; in SetLiteGraph() 192 m_outputIndices = {6};
|