| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | transpose_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
39 const std::vector<uint32_t>& outputsIndex) in InitTensor()
45 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
56 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
58 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
72 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> outputsIndex = { 3 }; in HWTEST_F() local
91 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
105 std::vector<uint32_t> outputsIndex in HWTEST_F() local
38 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
121 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
136 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
152 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
182 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
[all...] |
| H A D | tanh_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
39 const std::vector<uint32_t>& outputsIndex) in InitTensor()
45 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
56 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
58 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
72 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
91 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
105 std::vector<uint32_t> outputsIndex in HWTEST_F() local
38 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
121 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
136 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
153 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
183 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
[all...] |
| H A D | sub_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
52 void SubBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor() argument
60 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
71 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
92 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
108 std::vector<uint32_t> outputsIndex = { 4 }; in HWTEST_F() local
111 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
126 std::vector<uint32_t> outputsIndex in HWTEST_F() local
144 std::vector<uint32_t> outputsIndex = { 3 }; HWTEST_F() local
159 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
177 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
195 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
213 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
235 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
253 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
271 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
307 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
[all...] |
| H A D | unsqueeze_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
52 const std::vector<uint32_t>& outputsIndex) in InitTensor()
60 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
71 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
74 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
89 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
92 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
108 std::vector<uint32_t> outputsIndex = { 3 }; in HWTEST_F() local
111 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
126 std::vector<uint32_t> outputsIndex in HWTEST_F() local
51 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
144 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
159 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
177 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
195 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
213 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
234 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
252 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
282 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
[all...] |
| H A D | stack_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
42 const std::vector<uint32_t>& outputsIndex) in InitTensor()
50 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
72 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
75 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
90 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
93 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
109 std::vector<uint32_t> outputsIndex = { 3 }; in HWTEST_F() local
112 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
127 std::vector<uint32_t> outputsIndex in HWTEST_F() local
41 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
145 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
160 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
178 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
196 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
214 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
235 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
253 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
283 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
[all...] |
| H A D | softmax_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
53 const std::vector<uint32_t>& outputsIndex) in InitTensor()
61 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
72 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
75 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
90 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
93 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
109 std::vector<uint32_t> outputsIndex = { 3 }; in HWTEST_F() local
112 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
127 std::vector<uint32_t> outputsIndex in HWTEST_F() local
52 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
145 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
159 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
177 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
198 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
216 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
234 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
254 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
284 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
[all...] |
| H A D | squared_difference_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
40 const std::vector<uint32_t>& outputsIndex) in InitTensor()
46 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
64 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
69 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
82 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
87 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
101 std::vector<uint32_t> outputsIndex = { 3 }; in HWTEST_F() local
106 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
119 std::vector<uint32_t> outputsIndex in HWTEST_F() local
39 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
137 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
151 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
168 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
197 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
[all...] |
| H A D | sqrt_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
40 const std::vector<uint32_t>& outputsIndex) in InitTensor()
46 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
64 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
69 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
82 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
87 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
101 std::vector<uint32_t> outputsIndex = { 2 }; in HWTEST_F() local
106 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
119 std::vector<uint32_t> outputsIndex in HWTEST_F() local
39 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
137 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
151 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
169 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
198 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
[all...] |
| H A D | split_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
47 void SplitBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor() argument
55 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
100 std::vector<uint32_t> outputsIndex = { 1, 2 }; in HWTEST_F() local
103 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
120 std::vector<uint32_t> outputsIndex = { 1, 2 }; in HWTEST_F() local
123 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
141 std::vector<uint32_t> outputsIndex = { 2, 3 }; in HWTEST_F() local
144 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
161 std::vector<uint32_t> outputsIndex in HWTEST_F() local
182 std::vector<uint32_t> outputsIndex = { 4, 5 }; HWTEST_F() local
196 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
216 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
236 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
256 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
276 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
296 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
319 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
338 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
358 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
378 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
408 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
[all...] |
| H A D | squeeze_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
53 const std::vector<uint32_t>& outputsIndex) in InitTensor()
61 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
72 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
75 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
90 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
93 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
109 std::vector<uint32_t> outputsIndex = { 3 }; in HWTEST_F() local
112 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
127 std::vector<uint32_t> outputsIndex in HWTEST_F() local
52 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
145 std::vector<uint32_t> outputsIndex = { 2 }; HWTEST_F() local
160 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
178 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
196 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
217 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
247 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
[all...] |
| H A D | spacetobatchnd_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
37 void InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex,
47 const std::vector<uint32_t>& outputsIndex) in InitTensor()
55 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
59 const std::vector<uint32_t>& outputsIndex, const std::vector<uint32_t>& paramsIndex) in InitTensor()
68 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
119 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
123 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
138 std::vector<uint32_t> outputsIndex = { 1 }; in HWTEST_F() local
142 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
46 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
58 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<uint32_t>& paramsIndex) InitTensor() argument
160 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
180 std::vector<uint32_t> outputsIndex = { 1, 2 }; HWTEST_F() local
200 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
220 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
241 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
262 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
278 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
294 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
310 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
334 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
358 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
378 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
398 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
418 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
457 std::vector<uint32_t> outputsIndex = { 1 }; HWTEST_F() local
[all...] |
| H A D | topk_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
43 void TopKBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor() argument
49 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
81 std::vector<uint32_t> outputsIndex = { 2, 3 }; in HWTEST_F() local
85 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
101 std::vector<uint32_t> outputsIndex = { 2, 3 }; in HWTEST_F() local
105 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
122 std::vector<uint32_t> outputsIndex = { 4, 5 }; in HWTEST_F() local
126 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
142 std::vector<uint32_t> outputsIndex in HWTEST_F() local
162 std::vector<uint32_t> outputsIndex = { 3, 4 }; HWTEST_F() local
177 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
197 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
217 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
240 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
263 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
283 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
315 std::vector<uint32_t> outputsIndex = { 2, 3 }; HWTEST_F() local
[all...] |
| H A D | slice_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
39 std::vector<uint32_t> outputsIndex = { 3 }; member in OHOS::NeuralNetworkRuntime::UnitTest::SliceBuilderTest
45 const std::vector<uint32_t>& outputsIndex) in InitTensor()
52 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
72 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
86 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
102 outputsIndex = { 4 }; in HWTEST_F()
105 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
120 outputsIndex = { 3, 4 }; in HWTEST_F()
123 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
44 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
[all...] |
| H A D | tile_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
39 std::vector<uint32_t> outputsIndex = { 2 }; member in OHOS::NeuralNetworkRuntime::UnitTest::TileBuilderTest
45 const std::vector<uint32_t>& outputsIndex) in InitTensor()
52 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
72 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
86 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
102 outputsIndex = { 3 }; in HWTEST_F()
105 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
119 outputsIndex = { 2, 3 }; in HWTEST_F()
122 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
44 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
[all...] |
| H A D | strided_slice_builder_test.cpp | 32 const std::vector<uint32_t>& outputsIndex) override;
110 const std::vector<uint32_t>& outputsIndex) in InitTensor()
118 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in InitTensor()
139 std::vector<uint32_t> outputsIndex = { 4 }; in HWTEST_F() local
141 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
156 std::vector<uint32_t> outputsIndex = { 4 }; in HWTEST_F() local
158 InitTensor(inputsIndex, outputsIndex); in HWTEST_F()
174 std::vector<uint32_t> outputsIndex = { 10 }; in HWTEST_F() local
181 SaveOutputTensor(outputsIndex, OH_NN_FLOAT32, OutputDim, nullptr); in HWTEST_F()
196 std::vector<uint32_t> outputsIndex in HWTEST_F() local
109 InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) InitTensor() argument
218 std::vector<uint32_t> outputsIndex = { 9 }; HWTEST_F() local
233 std::vector<uint32_t> outputsIndex = {}; HWTEST_F() local
255 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
282 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
303 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
324 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
345 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
367 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
393 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
420 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
447 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
474 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
500 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
534 std::vector<uint32_t> outputsIndex = { 4 }; HWTEST_F() local
[all...] |
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | ops_builder.cpp | 38 void OpsBuilder::GetOutputIndex(std::vector<uint32_t>& outputsIndex, in GetOutputIndex() argument
42 std::transform(m_outputsIndex.begin(), m_outputsIndex.end(), std::back_inserter(outputsIndex), in GetOutputIndex()
59 const std::vector<uint32_t>& outputsIndex, in CheckIOIndex()
65 size_t outputIndexSize = outputsIndex.size(); in CheckIOIndex()
84 isOverTensorSize = std::any_of(outputsIndex.begin(), outputsIndex.end(), [allTensorsSize](uint32_t index) { in CheckIOIndex()
117 void OpsBuilder::SetQuantType(const std::vector<uint32_t>& outputsIndex, in SetQuantType() argument
120 if (allTensors[outputsIndex.front()]->IsQuantTensor()) { in SetQuantType()
58 CheckIOIndex(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors, const size_t inputNum, const size_t outputNum) const CheckIOIndex() argument
|
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | concat_builder.cpp | 56 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build()
69 if (outputsIndex.size() != OUTPUT_NUM) { in Build()
74 OH_NN_ReturnCode returnCode = SetInputsAndOutputs(inputsIndex, outputsIndex, allTensors); in Build()
102 SetQuantType(outputsIndex, allTensors); in Build()
110 const std::vector<uint32_t>& outputsIndex, in SetInputsAndOutputs()
122 isOverTensorSize = std::any_of(outputsIndex.begin(), outputsIndex.end(), [allTensorsSize](uint32_t index) { in SetInputsAndOutputs()
134 m_outputsIndex = outputsIndex; 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
109 SetInputsAndOutputs(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, const std::vector<std::shared_ptr<NNTensor>>& allTensors) SetInputsAndOutputs() argument
|
| H A D | hswish_builder.cpp | 35 const std::vector<uint32_t>& outputsIndex, in Build()
43 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUMS, OUTPUT_NUMS); in Build()
56 m_outputsIndex = outputsIndex; in Build()
59 SetQuantType(outputsIndex, allTensors); 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
|
| H A D | relu6_builder.cpp | 36 const std::vector<uint32_t>& outputsIndex, in Build()
44 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
51 m_outputsIndex = outputsIndex; in Build()
59 SetQuantType(outputsIndex, allTensors); 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 | tanh_builder.cpp | 38 const std::vector<uint32_t>& outputsIndex, in Build()
46 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
53 m_outputsIndex = outputsIndex; in Build()
62 SetQuantType(outputsIndex, allTensors); in Build()
36 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 | 36 const std::vector<uint32_t>& outputsIndex, in Build()
43 auto ret = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
49 m_outputsIndex = outputsIndex; in Build()
69 SetQuantType(outputsIndex, allTensors); 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 | bias_add_builder.cpp | 32 const std::vector<uint32_t>& outputsIndex, in Build()
40 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
47 m_outputsIndex = outputsIndex; in Build()
56 SetQuantType(outputsIndex, allTensors); in Build()
30 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 | hard_sigmoid_builder.cpp | 35 const std::vector<uint32_t>& outputsIndex,
in Build()
43 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUMS, OUTPUT_NUMS);
in Build()
56 m_outputsIndex = outputsIndex;
in Build()
59 SetQuantType(outputsIndex, allTensors);
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
|
| H A D | greater_equal_builder.cpp | 35 const std::vector<uint32_t>& outputsIndex,
in Build()
43 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUMS, OUTPUT_NUMS);
in Build()
56 m_outputsIndex = outputsIndex;
in Build()
59 SetQuantType(outputsIndex, allTensors);
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
|
| H A D | gather_nd_builder.cpp | 35 const std::vector<uint32_t>& outputsIndex,
in Build()
43 OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM);
in Build()
56 m_outputsIndex = outputsIndex;
in Build()
59 SetQuantType(outputsIndex, allTensors);
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
|