| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | transpose_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
38 void TransposeBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
44 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
55 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
58 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
71 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
88 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; in HWTEST_F() local
91 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
104 std::vector<uint32_t> inputsIndex in HWTEST_F() local
120 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
135 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
151 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
181 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
[all...] |
| H A D | tanh_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
38 void TanhBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
44 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
55 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
58 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
71 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
88 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
91 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
104 std::vector<uint32_t> inputsIndex in HWTEST_F() local
120 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
135 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
152 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
182 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | sub_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
52 void SubBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor() argument
59 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
70 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
88 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
92 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
107 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; in HWTEST_F() local
111 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
125 std::vector<uint32_t> inputsIndex in HWTEST_F() local
143 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; HWTEST_F() local
158 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; HWTEST_F() local
176 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
194 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
212 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
234 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
252 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
270 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
306 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
[all...] |
| H A D | unsqueeze_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
51 void UnsqueezeBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
59 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
70 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
88 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
92 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
107 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; in HWTEST_F() local
111 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
125 std::vector<uint32_t> inputsIndex in HWTEST_F() local
143 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
158 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
176 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
194 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
212 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
233 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
251 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
281 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | stack_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
41 void StackBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
49 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
71 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
75 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
93 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
108 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; in HWTEST_F() local
112 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
126 std::vector<uint32_t> inputsIndex in HWTEST_F() local
144 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
159 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
177 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
195 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
213 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
234 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
252 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
282 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
[all...] |
| H A D | softmax_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
52 void SoftmaxBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
60 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
71 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
75 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
93 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
108 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; in HWTEST_F() local
112 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
126 std::vector<uint32_t> inputsIndex in HWTEST_F() local
144 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
158 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
176 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
197 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
215 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
233 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
253 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
283 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | squared_difference_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
39 void SquaredDifferenceBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
45 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
63 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
68 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
81 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
86 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
100 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; in HWTEST_F() local
105 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
118 std::vector<uint32_t> inputsIndex in HWTEST_F() local
136 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
150 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
167 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
196 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
[all...] |
| H A D | sqrt_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
39 void SqrtBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
45 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
63 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
68 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
81 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
86 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
100 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
105 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
118 std::vector<uint32_t> inputsIndex in HWTEST_F() local
136 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
150 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
168 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
197 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | split_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
47 void SplitBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor() argument
54 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
99 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
103 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
119 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
123 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
140 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
144 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
160 std::vector<uint32_t> inputsIndex in HWTEST_F() local
181 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
195 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
215 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
235 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
255 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
275 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
295 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
318 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
337 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
357 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
377 std::vector<uint32_t> inputsIndex = { 1 }; HWTEST_F() local
407 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | squeeze_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
52 void SqueezeBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
60 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
71 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
75 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
93 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
108 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; in HWTEST_F() local
112 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
126 std::vector<uint32_t> inputsIndex in HWTEST_F() local
144 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
159 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
177 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
195 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
216 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
246 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | spacetobatchnd_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
37 void InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex,
46 void SpaceToBatchNDBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
54 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
58 void SpaceToBatchNDBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
67 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
118 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
123 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
137 std::vector<uint32_t> inputsIndex = { 0 }; in HWTEST_F() local
142 InitTensor(inputsIndex, outputsInde in HWTEST_F()
159 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
179 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
199 std::vector<uint32_t> inputsIndex = {}; HWTEST_F() local
219 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
240 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
261 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
277 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
293 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
309 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
333 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
357 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
377 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
397 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
417 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
456 std::vector<uint32_t> inputsIndex = { 0 }; HWTEST_F() local
[all...] |
| H A D | topk_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
43 void TopKBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor() argument
48 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
80 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
85 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
100 std::vector<uint32_t> inputsIndex = { 0, 1 }; in HWTEST_F() local
105 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
121 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; in HWTEST_F() local
126 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
141 std::vector<uint32_t> inputsIndex in HWTEST_F() local
161 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; HWTEST_F() local
176 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
196 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
216 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
239 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
262 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
282 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
314 std::vector<uint32_t> inputsIndex = { 0, 1 }; HWTEST_F() local
[all...] |
| H A D | slice_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
38 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; member in OHOS::NeuralNetworkRuntime::UnitTest::SliceBuilderTest
44 void SliceBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
51 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
72 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
86 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
101 inputsIndex = { 0, 1, 2, 3 }; in HWTEST_F()
105 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
119 inputsIndex = { 0, 1, 2 }; in HWTEST_F()
123 InitTensor(inputsIndex, outputsInde in HWTEST_F()
[all...] |
| H A D | strided_slice_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
109 void StridedSliceBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
117 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
138 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; in HWTEST_F() local
141 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
155 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; in HWTEST_F() local
158 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
173 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }; in HWTEST_F() local
180 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in HWTEST_F()
195 std::vector<uint32_t> inputsIndex in HWTEST_F() local
217 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3, 4, 5, 6, 7, 8 }; HWTEST_F() local
232 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
254 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
281 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
302 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
323 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
344 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
366 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
392 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
419 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
446 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
473 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
499 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
533 std::vector<uint32_t> inputsIndex = { 0, 1, 2, 3 }; HWTEST_F() local
[all...] |
| H A D | tile_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
38 std::vector<uint32_t> inputsIndex = { 0, 1 }; member in OHOS::NeuralNetworkRuntime::UnitTest::TileBuilderTest
44 void TileBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, in InitTensor() argument
51 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
72 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
86 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
101 inputsIndex = { 0, 1, 2 }; in HWTEST_F()
105 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
122 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
136 OH_NN_ReturnCode ret = m_builder.Build(paramsIndex, inputsIndex, outputsInde in HWTEST_F()
[all...] |
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | layernorm_builder.cpp | 106 const std::vector<uint32_t>& inputsIndex, in Build()
115 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
121 m_inputsIndex = inputsIndex; in Build()
145 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build()
153 returnCode = ValidateGammaAndBetaShape(inputsIndex, m_beginNormAxis, allTensors); in Build()
179 OH_NN_ReturnCode LayerNormBuilder::ValidateGammaAndBetaShape(const std::vector<uint32_t>& inputsIndex, in ValidateGammaAndBetaShape() argument
182 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in ValidateGammaAndBetaShape()
183 auto gammaShape = allTensors[inputsIndex[INPUT_GAMMA]]->GetDimensions(); in ValidateGammaAndBetaShape()
184 auto betaShape = allTensors[inputsIndex[INPUT_BETA]]->GetDimensions(); in ValidateGammaAndBetaShape()
105 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| H A D | conv2d_builder.cpp | 43 OH_NN_ReturnCode Conv2DBuilder::SetInputAndOutput(const std::vector<uint32_t>& inputsIndex, in SetInputAndOutput() argument
47 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in SetInputAndOutput()
53 m_inputsIndex = inputsIndex; in SetInputAndOutput()
59 OH_NN_ReturnCode Conv2DBuilder::SetChannel(const std::vector<uint32_t>& inputsIndex, in SetChannel() argument
63 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetChannel()
75 void Conv2DBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() argument
79 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
227 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build()
235 OH_NN_ReturnCode returnCode = SetInputAndOutput(inputsIndex, outputsIndex, allTensors); in Build()
246 returnCode = SetChannel(inputsIndex, allTensor in Build()
226 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
[all...] |
| H A D | concat_builder.cpp | 56 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build()
64 if (inputsIndex.size() < MINIMUM_INTPUT) { in Build()
74 OH_NN_ReturnCode returnCode = SetInputsAndOutputs(inputsIndex, outputsIndex, allTensors); in Build()
109 OH_NN_ReturnCode ConcatBuilder::SetInputsAndOutputs(const std::vector<uint32_t>& inputsIndex, in SetInputsAndOutputs() argument
114 bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint32_t index) { in SetInputsAndOutputs()
131 m_inputsIndex = inputsIndex; in SetInputsAndOutputs()
55 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| H A D | cast_builder.cpp | 35 const std::vector<uint32_t>& inputsIndex, in Build()
43 auto ret = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
48 m_inputsIndex = inputsIndex; in Build()
57 auto castType = allTensors[inputsIndex[INPUT_TYPE]]->GetBuffer(); in Build()
34 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| H A D | depthwise_conv2d_native_builder.cpp | 87 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() argument
91 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
185 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in SetInputAndOutput()
188 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in SetInputAndOutput()
194 m_inputsIndex = inputsIndex; in SetInputAndOutput()
201 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build()
209 OH_NN_ReturnCode ret = SetInputAndOutput(inputsIndex, outputsIndex, allTensors); in Build()
220 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build()
227 ret = SetKernelSize(inputsIndex, allTensors); in Build()
184 SetInputAndOutput( const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) SetInputAndOutput() argument
200 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| H A D | split_builder.cpp | 29 OH_NN_ReturnCode SplitBuilder::SetInputAndOutput(const std::vector<uint32_t> &inputsIndex, in SetInputAndOutput() argument
32 auto inputSize = inputsIndex.size(); in SetInputAndOutput()
34 LOGE("[SplitBuilder] The number of inputsIndex should be %d, its number is %zu.", INPUT_NUM, inputSize); in SetInputAndOutput()
39 bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorSize](uint32_t index) { in SetInputAndOutput()
55 m_inputsIndex = inputsIndex; in SetInputAndOutput()
125 const std::vector<uint32_t> &inputsIndex, in Build()
134 OH_NN_ReturnCode returnCode = SetInputAndOutput(inputsIndex, outputsIndex, allTensors); in Build()
124 Build(const std::vector<uint32_t> ¶msIndex, const std::vector<uint32_t> &inputsIndex, const std::vector<uint32_t> &outputsIndex, const std::vector<std::shared_ptr<NNTensor>> &allTensors) Build() argument
|
| H A D | fullconnection_builder.cpp | 33 OH_NN_ReturnCode FullConnectionBuilder::SetFullConnectionInput(const std::vector<uint32_t>& inputsIndex, in SetFullConnectionInput() argument
43 bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint32_t index) { in SetFullConnectionInput()
51 m_inputsIndex = inputsIndex; in SetFullConnectionInput()
171 const std::vector<uint32_t>& inputsIndex, in Build()
180 OH_NN_ReturnCode returnCode = SetFullConnectionInput(inputsIndex, outputsIndex, allTensors); in Build()
170 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| H A D | unstack_builder.cpp | 54 const std::vector<uint32_t>& inputsIndex,
in Build()
63 if (inputsIndex.size() != INPUT_NUM) {
in Build()
74 bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint32_t index) {
in Build()
90 m_inputsIndex = inputsIndex;
in Build()
53 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| H A D | hswish_builder.cpp | 34 const std::vector<uint32_t>& inputsIndex, in Build()
43 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUMS, OUTPUT_NUMS); in Build()
55 m_inputsIndex = inputsIndex; in Build()
33 Build(const std::vector<uint32_t>& paramsIndex, const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) Build() argument
|
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | ops_builder.cpp | 28 void OpsBuilder::GetInputIndex(std::vector<uint32_t>& inputsIndex, in GetInputIndex() argument
32 std::transform(m_inputsIndex.begin(), m_inputsIndex.end(), std::back_inserter(inputsIndex), in GetInputIndex()
58 OH_NN_ReturnCode OpsBuilder::CheckIOIndex(const std::vector<uint32_t>& inputsIndex, in CheckIOIndex() argument
64 size_t inputsIndexSize = inputsIndex.size(); in CheckIOIndex()
76 bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint32_t index) { in CheckIOIndex()
|