Home
last modified time | relevance | path

Searched refs:paramDim (Results 1 - 25 of 28) sorted by relevance

12

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dsplit_builder_test.cpp101 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
105 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F()
106 SaveOutputNumTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_OUTPUT_NUM); in HWTEST_F()
107 SaveSizeSplitsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_SIZE_SPLITS); in HWTEST_F()
121 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
124 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F()
125 SaveOutputNumTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_OUTPUT_NUM); in HWTEST_F()
126 SaveSizeSplitsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_SIZE_SPLITS); in HWTEST_F()
142 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
145 SaveAxisTensor(OH_NN_INT64, paramDim, nullpt in HWTEST_F()
162 std::vector<int32_t> paramDim = {}; HWTEST_F() local
197 std::vector<int32_t> paramDim = {}; HWTEST_F() local
217 std::vector<int32_t> paramDim = {}; HWTEST_F() local
237 std::vector<int32_t> paramDim = {}; HWTEST_F() local
257 std::vector<int32_t> paramDim = {}; HWTEST_F() local
277 std::vector<int32_t> paramDim = {}; HWTEST_F() local
297 std::vector<int32_t> paramDim = {}; HWTEST_F() local
320 std::vector<int32_t> paramDim = {}; HWTEST_F() local
339 std::vector<int32_t> paramDim = {}; HWTEST_F() local
359 std::vector<int32_t> paramDim = {}; HWTEST_F() local
409 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dstrided_slice_builder_test.cpp123 std::vector<int32_t> paramDim = {}; in InitParams() local
124 SaveBeginMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_BEGIN_MASK); in InitParams()
125 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); in InitParams()
126 SaveEllipsisMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_ELLIPSIS_MASK); in InitParams()
127 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK); in InitParams()
128 SaveShrinkAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_SHRINK_AXIS_MASK); in InitParams()
259 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
264 SaveBeginMaskTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_STRIDED_SLICE_BEGIN_MASK); in HWTEST_F()
265 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); in HWTEST_F()
266 SaveEllipsisMaskTensor(OH_NN_INT64, paramDim, nullpt in HWTEST_F()
283 std::vector<int32_t> paramDim = {}; HWTEST_F() local
304 std::vector<int32_t> paramDim = {}; HWTEST_F() local
325 std::vector<int32_t> paramDim = {}; HWTEST_F() local
346 std::vector<int32_t> paramDim = {}; HWTEST_F() local
368 std::vector<int32_t> paramDim = {}; HWTEST_F() local
394 std::vector<int32_t> paramDim = {}; HWTEST_F() local
421 std::vector<int32_t> paramDim = {}; HWTEST_F() local
448 std::vector<int32_t> paramDim = {}; HWTEST_F() local
475 std::vector<int32_t> paramDim = {}; HWTEST_F() local
501 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dsub_builder_test.cpp72 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
75 SaveActivateTypeTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_SUB_ACTIVATIONTYPE); in HWTEST_F()
90 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
93 SaveActivateTypeTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_SUB_ACTIVATIONTYPE); in HWTEST_F()
109 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
112 SaveActivateTypeTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_SUB_ACTIVATIONTYPE); in HWTEST_F()
127 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
130 SaveActivateTypeTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_SUB_ACTIVATIONTYPE); in HWTEST_F()
160 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
163 SaveActivateTypeTensor(OH_NN_BOOL, paramDim, nullpt in HWTEST_F()
178 std::vector<int32_t> paramDim = {}; HWTEST_F() local
196 std::vector<int32_t> paramDim = {}; HWTEST_F() local
214 std::vector<int32_t> paramDim = {}; HWTEST_F() local
236 std::vector<int32_t> paramDim = {1, 2}; HWTEST_F() local
254 std::vector<int32_t> paramDim = {}; HWTEST_F() local
272 std::vector<int32_t> paramDim = {}; HWTEST_F() local
308 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dtopk_builder_test.cpp83 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
86 SaveSortedTensor(OH_NN_BOOL, paramDim, nullptr, OH_NN_TOP_K_SORTED); in HWTEST_F()
87 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F()
103 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
106 SaveSortedTensor(OH_NN_BOOL, paramDim, nullptr, OH_NN_TOP_K_SORTED); in HWTEST_F()
107 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F()
124 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
127 SaveSortedTensor(OH_NN_BOOL, paramDim, nullptr, OH_NN_TOP_K_SORTED); in HWTEST_F()
128 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F()
144 std::vector<int32_t> paramDim in HWTEST_F() local
179 std::vector<int32_t> paramDim = {}; HWTEST_F() local
199 std::vector<int32_t> paramDim = {}; HWTEST_F() local
219 std::vector<int32_t> paramDim = {}; HWTEST_F() local
242 std::vector<int32_t> paramDim = {}; HWTEST_F() local
265 std::vector<int32_t> paramDim = {}; HWTEST_F() local
285 std::vector<int32_t> paramDim = {}; HWTEST_F() local
317 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dunsqueeze_builder_test.cpp72 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
75 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F()
90 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
93 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F()
109 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
112 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F()
127 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
130 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F()
160 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
163 SaveAxisTensor(OH_NN_INT64, paramDim, nullpt in HWTEST_F()
178 std::vector<int32_t> paramDim = {}; HWTEST_F() local
196 std::vector<int32_t> paramDim = {}; HWTEST_F() local
214 std::vector<int32_t> paramDim = {}; HWTEST_F() local
235 std::vector<int32_t> paramDim = {1, 2}; HWTEST_F() local
253 std::vector<int32_t> paramDim = {}; HWTEST_F() local
283 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dstack_builder_test.cpp73 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F()
91 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F()
110 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F()
128 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F()
161 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
164 SaveAxisTensor(OH_NN_INT64, paramDim, nullpt in HWTEST_F()
179 std::vector<int32_t> paramDim = {}; HWTEST_F() local
197 std::vector<int32_t> paramDim = {}; HWTEST_F() local
215 std::vector<int32_t> paramDim = {}; HWTEST_F() local
236 std::vector<int32_t> paramDim = {1, 2}; HWTEST_F() local
254 std::vector<int32_t> paramDim = {}; HWTEST_F() local
284 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dsoftmax_builder_test.cpp73 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F()
91 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F()
110 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F()
128 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F()
160 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
163 SaveAxisTensor(OH_NN_INT64, paramDim, nullpt in HWTEST_F()
199 std::vector<int32_t> paramDim = {}; HWTEST_F() local
217 std::vector<int32_t> paramDim = {1, 5, 1, 1}; HWTEST_F() local
235 std::vector<int32_t> paramDim = {}; HWTEST_F() local
255 std::vector<int32_t> paramDim = {}; HWTEST_F() local
285 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dsqueeze_builder_test.cpp73 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F()
91 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F()
110 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F()
128 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F()
161 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
164 SaveAxisTensor(OH_NN_INT64, paramDim, nullpt in HWTEST_F()
179 std::vector<int32_t> paramDim = {}; HWTEST_F() local
197 std::vector<int32_t> paramDim = {}; HWTEST_F() local
218 std::vector<int32_t> paramDim = {}; HWTEST_F() local
248 std::vector<int32_t> paramDim = {}; HWTEST_F() local
[all...]
H A Dtile_builder_test.cpp41 std::vector<int32_t> paramDim = {}; member in OHOS::NeuralNetworkRuntime::UnitTest::TileBuilderTest
73 SaveDimsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TILE_DIMS); in HWTEST_F()
87 SaveDimsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TILE_DIMS); in HWTEST_F()
106 SaveDimsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TILE_DIMS); in HWTEST_F()
123 SaveDimsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TILE_DIMS); in HWTEST_F()
169 std::shared_ptr<NNTensor> dimsTensor = TransToNNTensor(OH_NN_FLOAT32, paramDim, in HWTEST_F()
192 SaveDimsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE); in HWTEST_F()
212 std::shared_ptr<NNTensor> dimsTensor = TransToNNTensor(OH_NN_INT64, paramDim, in HWTEST_F()
241 SaveDimsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TILE_DIMS); in HWTEST_F()
H A Dgreater_equal_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dgreater_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dgather_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dgather_nd_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dnot_equal_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dhard_sigmoid_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dhswish_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dlessequal_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dequal_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dfill_builder_test.cpp139 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
144 paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Drelu6_builder_test.cpp136 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
140 std::shared_ptr<NNTensor> relu6Tensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Drelu_builder_test.cpp136 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
140 std::shared_ptr<NNTensor> reluTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dreshape_builder_test.cpp137 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
141 std::shared_ptr<NNTensor> paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dprelu_builder_test.cpp137 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
141 std::shared_ptr<NNTensor> preluTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dsigmoid_builder_test.cpp136 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
140 std::shared_ptr<NNTensor> sigmoidTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
H A Dshape_builder_test.cpp138 std::vector<int32_t> paramDim = {}; in HWTEST_F() local
142 std::shared_ptr<NNTensor> paramTensor = TransToNNTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TENSOR); in HWTEST_F()

Completed in 10 milliseconds

12