Home
last modified time | relevance | path

Searched refs:paramsIndex (Results 1 - 25 of 242) sorted by relevance

12345678910

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dtopk_builder_test.cpp82 std::vector<uint32_t> paramsIndex = { 4, 5 }; in HWTEST_F() local
89 OH_NN_ReturnCode ret = m_builder.Build(paramsIndex, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
102 std::vector<uint32_t> paramsIndex = { 4, 5 }; in HWTEST_F() local
109 EXPECT_EQ(OH_NN_SUCCESS, m_builder.Build(paramsIndex, m_inputsIndex, m_outputsIndex, m_allTensors)); in HWTEST_F()
110 OH_NN_ReturnCode ret = m_builder.Build(paramsIndex, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
123 std::vector<uint32_t> paramsIndex = { 6, 7 }; in HWTEST_F() local
130 OH_NN_ReturnCode ret = m_builder.Build(paramsIndex, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
143 std::vector<uint32_t> paramsIndex = { 5, 6 }; in HWTEST_F() local
150 OH_NN_ReturnCode ret = m_builder.Build(paramsIndex, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
163 std::vector<uint32_t> paramsIndex in HWTEST_F() local
178 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
198 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
218 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
241 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
264 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
284 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
316 std::vector<uint32_t> paramsIndex = { 4, 5 }; HWTEST_F() local
[all...]
H A Dslice_builder_test.cpp40 std::vector<uint32_t> paramsIndex = { 4 }; member in OHOS::NeuralNetworkRuntime::UnitTest::SliceBuilderTest
50 m_paramsIndex = paramsIndex; in InitTensor()
103 paramsIndex = { 5 }; in HWTEST_F()
121 paramsIndex = { 5 }; in HWTEST_F()
139 paramsIndex = {}; in HWTEST_F()
157 paramsIndex = {}; in HWTEST_F()
177 m_paramsIndex = paramsIndex; in HWTEST_F()
203 m_paramsIndex = paramsIndex; in HWTEST_F()
223 m_paramsIndex = paramsIndex; in HWTEST_F()
H A Dtile_builder_test.cpp40 std::vector<uint32_t> paramsIndex = { 3 }; member in OHOS::NeuralNetworkRuntime::UnitTest::TileBuilderTest
50 m_paramsIndex = paramsIndex; in InitTensor()
103 paramsIndex = { 4 }; in HWTEST_F()
120 paramsIndex = { 4 }; in HWTEST_F()
136 OH_NN_ReturnCode ret = m_builder.Build(paramsIndex, inputsIndex, outputsIndex, m_allTensors); in HWTEST_F()
165 m_paramsIndex = paramsIndex; in HWTEST_F()
189 m_paramsIndex = paramsIndex; in HWTEST_F()
208 m_paramsIndex = paramsIndex; in HWTEST_F()
/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Derf_builder.cpp30 OH_NN_ReturnCode ErfBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
46 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_NUM); in Build()
52 if (!paramsIndex.empty()) { in Build()
53 LOGW("[Erf] Build failed, the erf expects no parameters, but receive %zu", paramsIndex.size()); in Build()
H A Dinstance_norm_builder.cpp53 OH_NN_ReturnCode InstanceNormBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
72 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
78 for (int i : paramsIndex) { in Build()
H A Donehot_builder.cpp52 OH_NN_ReturnCode OnehotBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
71 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
77 for (int i : paramsIndex) { in Build()
H A Dtile_builder.cpp63 OH_NN_ReturnCode TileBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
79 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
85 for (int i : paramsIndex) { in Build()
H A Dbroadcast_to_builder.cpp55 OH_NN_ReturnCode BroadcastToBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
74 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
80 for (int i : paramsIndex) { in Build()
H A Dbatchnorm_builder.cpp58 OH_NN_ReturnCode BatchNormBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
77 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
83 for (int i : paramsIndex) { in Build()
H A Dgelu_builder.cpp56 OH_NN_ReturnCode GeluBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
72 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
78 for (int i : paramsIndex) { in Build()
H A Dspace_to_depth_builder.cpp58 OH_NN_ReturnCode SpaceToDepthBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
77 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
83 for (int i : paramsIndex) { in Build()
H A Dadd_builder.cpp57 OH_NN_ReturnCode AddBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
76 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
82 for (uint32_t i : paramsIndex) { in Build()
H A Dflatten_builder.cpp53 OH_NN_ReturnCode FlattenBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
72 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
78 for (int i : paramsIndex) { in Build()
H A Dslice_builder.cpp63 OH_NN_ReturnCode SliceBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
79 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
85 for (int i : paramsIndex) { in Build()
H A Dsoftmax_builder.cpp57 OH_NN_ReturnCode SoftmaxBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
76 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
82 for (int i : paramsIndex) { in Build()
H A Dleaky_relu_builder.cpp53 OH_NN_ReturnCode LeakyReluBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
72 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
78 for (int i : paramsIndex) { in Build()
H A Dlog_softmax_builder.cpp56 OH_NN_ReturnCode LogSoftmaxBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
75 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
81 for (int i : paramsIndex) { in Build()
H A Dsqueeze_builder.cpp61 OH_NN_ReturnCode SqueezeBuilder::Build(const std::vector<uint32_t> &paramsIndex, in Build() argument
80 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
86 for (int i : paramsIndex) { in Build()
H A Dstack_builder.cpp60 OH_NN_ReturnCode StackBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
82 auto returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
88 for (int i : paramsIndex) { in Build()
H A Dassert_builder.cpp53 OH_NN_ReturnCode AssertBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
72 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
78 for (int i : paramsIndex) { in Build()
H A Dall_builder.cpp53 OH_NN_ReturnCode AllBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
72 ret = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
78 for (int i : paramsIndex) { in Build()
H A Dmul_builder.cpp65 OH_NN_ReturnCode MulBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
84 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
90 for (int i : paramsIndex) { in Build()
H A Dunsqueeze_builder.cpp57 OH_NN_ReturnCode UnsqueezeBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
75 returnCode = CheckParamIndex(paramsIndex, allTensors, PARAM_MAX_NUM); in Build()
81 for (int i : paramsIndex) { in Build()
H A Dmaxpool_builder.cpp27 OH_NN_ReturnCode MaxPoolBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build() argument
31 OH_NN_ReturnCode returnCode = PoolingBuild(paramsIndex, inputsIndex, outputsIndex, allTensors); in Build()
/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dops_builder.cpp95 OH_NN_ReturnCode OpsBuilder::CheckParamIndex(const std::vector<uint32_t>& paramsIndex, in CheckParamIndex() argument
99 size_t paramsIndexSize = paramsIndex.size(); in CheckParamIndex()
106 bool isParamsOutOfRange = std::any_of(paramsIndex.begin(), paramsIndex.end(), [allTensorsSize](uint32_t index) { in CheckParamIndex()

Completed in 14 milliseconds

12345678910