Home
last modified time | relevance | path

Searched refs:m_paramDim (Results 1 – 25 of 33) sorted by relevance

12

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlstm_test.cpp65 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::LSTMBuilderTest
233 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
258 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
288 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
316 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
370 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
425 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
455 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
485 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
515 SaveHasBias(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_LSTM_HAS_BIAS);
[all …]
H A Ddetection_post_process_test.cpp64 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::DetectionPostProcessBuilderTest
199 SetScale(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_SCALE);
223 SetScale(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_SCALE);
253 SetScale(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_SCALE);
358 std::shared_ptr<NNTensor> scaleTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
620 SetInputSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
646 SetScale(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
724 SetMaxDetections(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
802 SetNumClasses(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
828 SetUseRegularNms(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
[all …]
H A Dlrn_test.cpp51 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::LRNBuilderTest
119 SaveBeta(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BETA);
120 SaveBias(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BIAS);
138 SaveBeta(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BETA);
139 SaveBias(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BIAS);
162 SaveBeta(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BETA);
163 SaveBias(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BIAS);
184 SaveBeta(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BETA);
185 SaveBias(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BIAS);
232 SaveBeta(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LRN_BETA);
[all …]
H A Dresize_bilinear_builder_test.cpp58 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::ResizeBilinearBuilderTest
117 SaveHeightTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_HEIGHT); in SetParameterTensor()
118 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH); in SetParameterTensor()
233 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH);
281 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH);
306 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH);
331 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH);
336 m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_EXCLUDE_OUTSIDE);
356 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH);
494 SaveOutsideTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_SRC_T);
[all …]
H A Dl2_normalize_test.cpp45 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::L2NormalizeBuilderTest
81 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
98 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
120 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
140 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
204 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
226 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
264 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
281 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
319 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS);
[all …]
H A Dreduce_all_builder_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceAllBuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_ALL_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_ALL_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_ALL_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_ALL_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_ALL_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_ALL_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_ALL_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_ALL_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_ALL_COEFF);
[all …]
H A Dreduce_l2_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceL2BuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_L2_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_L2_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_L2_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_L2_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_L2_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_L2_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_L2_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_L2_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_L2_COEFF);
[all …]
H A Dreduce_max_builder_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceMaxBuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MAX_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MAX_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MAX_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MAX_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_MAX_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_MAX_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_MAX_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_MAX_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_MAX_COEFF);
[all …]
H A Dreduce_mean_builder_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceMeanBuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MEAN_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MEAN_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MEAN_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MEAN_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_MEAN_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_MEAN_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_MEAN_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_MEAN_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_MEAN_COEFF);
[all …]
H A Dreduce_min_builder_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceMinBuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MIN_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MIN_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MIN_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_MIN_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_MIN_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_MIN_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_MIN_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_MIN_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_MIN_COEFF);
[all …]
H A Dreduce_prod_builder_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceProdBuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_PROD_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_PROD_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_PROD_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_PROD_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_PROD_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_PROD_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_PROD_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_PROD_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_PROD_COEFF);
[all …]
H A Dreduce_sum_builder_test.cpp47 std::vector<int32_t> m_paramDim {1}; member in OHOS::NeuralNetworkRuntime::UnitTest::ReduceSumBuilderTest
94 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_SUM_COEFF);
111 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_SUM_COEFF);
133 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_SUM_COEFF);
153 SetCoeff(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_REDUCE_SUM_COEFF);
195 m_paramDim, nullptr, OH_NN_REDUCE_SUM_KEEP_DIMS);
219 m_paramDim, nullptr, OH_NN_REDUCE_SUM_COEFF);
243 m_paramDim, nullptr, OH_NN_REDUCE_SUM_REDUCE_TO_END);
390 m_paramDim, nullptr, OH_NN_REDUCE_SUM_KEEP_DIMS);
411 m_paramDim, nullptr, OH_NN_REDUCE_SUM_COEFF);
[all …]
H A Dexp_test.cpp46 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::ExpBuilderTest
92 SaveBase(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_BASE);
93 SaveScale(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_SCALE);
94 SaveShift(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_SHIFT);
109 SaveBase(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_BASE);
110 SaveScale(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_SCALE);
111 SaveShift(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_SHIFT);
131 SaveBase(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_BASE);
151 SaveBase(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_BASE);
216 SaveBase(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_EXP_BASE);
[all …]
H A Dquant_dtype_cast_builder_test.cpp46 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::QuantDTypeCastBuilderTest
92 SaveSrcTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_SRC_T);
93 SaveDstTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_DST_T);
94 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS);
109 SaveSrcTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_SRC_T);
110 SaveDstTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_DST_T);
111 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS);
131 SaveSrcTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_SRC_T);
262 SaveSrcTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
281 SaveDstTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
[all …]
H A Drange_test.cpp46 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::RangeBuilderTest
92 SaveStart(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_START);
93 SaveLimit(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_LIMIT);
94 SaveDelta(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_DELTA);
109 SaveStart(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_START);
110 SaveLimit(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_LIMIT);
111 SaveDelta(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_DELTA);
131 SaveStart(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_START);
132 SaveLimit(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_LIMIT);
133 SaveDelta(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_DELTA);
[all …]
H A Dmatmul_builder_test.cpp49 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::MatMulBuilderTest
104 SaveTransposeATensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_A);
105 SaveTransposeBTensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_B);
121 SaveTransposeATensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_A);
122 SaveTransposeBTensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_B);
143 SaveTransposeATensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_A);
144 SaveTransposeBTensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_B);
163 SaveTransposeATensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_A);
164 SaveTransposeBTensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_B);
203 SaveTransposeBTensor(OH_NN_BOOL, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_B);
[all …]
H A Dlayernorm_builder_test.cpp50 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::LayerNormBuilderTest
110 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
127 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
149 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
169 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
216 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
242 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
308 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
333 SaveEpsilonTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LAYER_NORM_EPSILON);
355 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
[all …]
H A Dpad_builder_test.cpp45 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::PadBuilderTest
82 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
98 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
119 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
138 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
182 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
218 m_paramDim = {2};
228 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
244 SetConstValueTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
245 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_PAD_PADDING_MODE);
[all …]
H A Dclip_test.cpp44 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::ClipBuilderTest
80 SaveMax(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MAX);
81 SaveMin(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MIN);
96 SaveMax(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MAX);
97 SaveMin(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MIN);
117 SaveMax(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MAX);
118 SaveMin(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MIN);
136 SaveMax(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MAX);
137 SaveMin(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MIN);
183 SaveMin(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_CLIP_MIN);
[all …]
H A Ddepth_to_space_test.cpp45 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::DepthToSpaceBuilderTest
82 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
98 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
119 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
138 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
184 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
221 SaveBlockSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
222 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
238 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
257 SaveMode(OH_NN_INT32, m_paramDim, nullptr, OH_NN_DEPTH_TO_SPACE_MODE);
[all …]
H A Dscale_builder_test.cpp44 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::ScaleBuilderTest
80 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
96 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
117 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
136 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
199 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
245 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
269 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
272 m_paramDim, nullptr, OH_NN_SCALE_ACTIVATIONTYPE);
291 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
[all …]
H A Dmul_builder_test.cpp43 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::MulBuilderTest
69 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
84 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
104 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
122 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
160 std::vector<int32_t> m_paramDim = {}; variable
183 m_paramDim = {2};
188 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, m_paramDim,
209 std::shared_ptr<NNTensor> activationTensor = TransToNNTensor(OH_NN_INT8, m_paramDim,
229 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MATMUL_ACTIVATION_TYPE);
[all …]
H A Dbatchnorm_builder_test.cpp43 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::BatchNormBuilderTest
69 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
84 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
104 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
122 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
161 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_INT32, m_paramDim,
179 std::vector<int32_t> m_paramDim = { 2 }; variable
183 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
203 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_A);
218 std::shared_ptr<NNTensor> epsilonTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, nullptr,
[all …]
H A Dall_test.cpp43 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::AllBuilderTest
81 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
96 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
116 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
134 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
174 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
193 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
209 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
226 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
H A Dassert_test.cpp43 std::vector<int32_t> m_paramDim {}; member in OHOS::NeuralNetworkRuntime::UnitTest::AssertBuilderTest
81 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
96 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
116 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
134 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
174 std::shared_ptr<NNTensor> summarizeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
193 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
209 std::shared_ptr<NNTensor> summarizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
226 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);

12