| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | conv2d_tranpose_padmode_test.cpp | 58 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
69 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetOutPaddings() local
229 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
288 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
347 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
376 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
489 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
518 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
543 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
648 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
[all …]
|
| H A D | fullconnection_with_axis_test.cpp | 79 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetActivation() local
90 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetAxis() local
101 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetUseAxis() local
112 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetHasBias() local
240 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
266 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
292 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
443 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
467 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_BOOL, m_param_dim, nullptr, variable
491 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_BOOL, m_param_dim, nullptr, variable
[all …]
|
| H A D | avgpool_pad_test.cpp | 60 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetRoundMode() local
70 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetGlobal() local
81 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
215 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
296 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
323 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, variable
350 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
378 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
467 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
491 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, variable
[all …]
|
| H A D | maxpool_pad_test.cpp | 60 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetRoundMode() local
70 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetGlobal() local
81 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
213 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
297 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
324 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, variable
351 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
379 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
467 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
492 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
[all …]
|
| H A D | conv2d_pad_test.cpp | 57 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
218 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_CONV2… variable
271 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, OH_NN_CONV2D_P… variable
298 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
351 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, groupDim, nullptr, OH_NN_CONV2D_GR… variable
379 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
400 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_stride_dim, nullptr, variable
425 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_dilation_dim, nullptr, variable
449 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_pad_dim, nullptr, variable
473 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, variable
[all …]
|
| H A D | argmax_test.cpp | 58 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetArgmaxAxis() local
68 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetArgmaxKeepdims() local
78 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetArgmaxTopK() local
88 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetArgmaxOutMaxValue() local
240 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_FLOAT32, m_param_dim, nullptr, OH_NN_ARG_… variable
265 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MA… variable
291 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_FLOAT32, m_param_dim, nullptr, OH_NN_ARG_… variable
316 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MA… variable
412 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MA… variable
433 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX… variable
[all …]
|
| H A D | avgpool_padmod_test.cpp | 59 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetRoundMode() local
69 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetGlobal() local
79 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
215 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
243 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_AVG_P… variable
298 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
325 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, variable
352 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
380 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
404 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
[all …]
|
| H A D | maxpool_padmode_test.cpp | 59 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetRoundMode() local
69 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetGlobal() local
79 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
209 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_kenelsize_dim, nullptr, variable
266 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_MAX_PO… variable
290 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
317 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, variable
344 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
372 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
396 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
[all …]
|
| H A D | conv2d_transpose_pad_test.cpp | 60 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
72 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetOutPaddings() local
96 std::shared_ptr<NNTensor> tensor; in SetConv2dTransposeInput() local
236 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
263 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, variable
290 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, OH_NN_CONV2D_T… variable
317 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_outpaddings_dim, nullptr, variable
344 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
370 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
395 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_outpaddings_dim, nullptr, variable
[all …]
|
| H A D | conv2d_padmode_test.cpp | 55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPadMode() local
216 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, OH_NN_CONV2… variable
244 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_dilation_dim, nullptr, OH_NN_CON… variable
276 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
303 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONV2D… variable
357 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, groupDim, nullptr, OH_NN_CONV2D_GR… variable
385 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
413 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
437 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_CONV2D_… variable
499 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
[all …]
|
| H A D | depthwise_conv2d_native_pad_test.cpp | 57 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetPad() local
219 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_stride_dim, nullptr, variable
274 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_pad_dim, nullptr, variable
300 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
327 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
353 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
379 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, activationDim, nullptr, variable
406 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_pad_dim, nullptr, variable
500 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_stride_dim, nullptr, variable
548 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_pad_dim, nullptr, variable
[all …]
|
| H A D | eltwise_test.cpp | 51 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetEltwiseMode() local
171 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_ELTWIS… variable
194 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ELTWISE… variable
215 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ELTWISE… variable
236 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACT… variable
254 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ELTWISE… variable
|
| H A D | fullconnection_test.cpp | 56 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_FLOAT32, m_input_dim, nullptr, OH_NN_TENS… in SetInputToAlltensor() local
78 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetActivation() local
89 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetHasBias() local
189 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
214 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, variable
238 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
262 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
286 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
|
| H A D | div_test.cpp | 53 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
166 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_DIV_AC… variable
186 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACT… variable
206 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACT… variable
225 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACT… variable
244 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, variable
|
| H A D | batch_to_space_nd_test.cpp | 55 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetBlockSize() local
67 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SetCrops() local
199 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_crops_dim, nullptr, variable
221 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_block_dim, nullptr, variable
244 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_block_dim, nullptr, variable
266 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_block_dim, nullptr, variable
286 std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_crops_dim, nullptr, variable
|
| H A D | add_test.cpp | 53 std::shared_ptr<NNTensor> tensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
172 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_ADD_AC… variable
191 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACT… variable
211 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ARG_MAX… variable
231 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACT… variable
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/ |
| H A D | nn_tensor_test.cpp | 40 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
53 OH_NN_Tensor tensor; variable
72 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
86 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
100 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
189 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
205 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
221 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
238 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
410 OH_NN_Tensor tensor {dataType, 2, dim, nullptr, OH_NN_TENSOR}; variable
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/ |
| H A D | nn_tensor_test.cpp | 40 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
53 OH_NN_Tensor tensor; variable
72 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
86 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
100 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
189 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
205 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
221 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
238 OH_NN_Tensor tensor {OH_NN_FLOAT32, 2, dim, nullptr, OH_NN_TENSOR}; variable
410 OH_NN_Tensor tensor {dataType, 2, dim, nullptr, OH_NN_TENSOR}; variable
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | detection_post_process_builder.cpp | 36 OH_NN_ReturnCode DetectionPostProcessBuilder::SetInputSize(const std::shared_ptr<NNTensor>& tensor) in SetInputSize()
58 OH_NN_ReturnCode DetectionPostProcessBuilder::SetScale(const std::shared_ptr<NNTensor>& tensor) in SetScale()
84 …ReturnCode DetectionPostProcessBuilder::SetNmsIoUThreshold(const std::shared_ptr<NNTensor>& tensor) in SetNmsIoUThreshold()
106 …turnCode DetectionPostProcessBuilder::SetNmsScoreThreshold(const std::shared_ptr<NNTensor>& tensor) in SetNmsScoreThreshold()
128 …N_ReturnCode DetectionPostProcessBuilder::SetMaxDetections(const std::shared_ptr<NNTensor>& tensor) in SetMaxDetections()
150 …urnCode DetectionPostProcessBuilder::SetDetectionsPerClass(const std::shared_ptr<NNTensor>& tensor) in SetDetectionsPerClass()
172 …ode DetectionPostProcessBuilder::SetMaxClassesPerDetection(const std::shared_ptr<NNTensor>& tensor) in SetMaxClassesPerDetection()
194 OH_NN_ReturnCode DetectionPostProcessBuilder::SetNumClasses(const std::shared_ptr<NNTensor>& tensor) in SetNumClasses()
216 …N_ReturnCode DetectionPostProcessBuilder::SetUseRegularNms(const std::shared_ptr<NNTensor>& tensor) in SetUseRegularNms()
238 …NN_ReturnCode DetectionPostProcessBuilder::SetOutQuantized(const std::shared_ptr<NNTensor>& tensor) in SetOutQuantized()
[all …]
|
| H A D | lstm_builder.cpp | 31 OH_NN_ReturnCode LSTMBuilder::SetBidirectional(const std::shared_ptr<NNTensor>& tensor) in SetBidirectional()
53 OH_NN_ReturnCode LSTMBuilder::SetHasBias(const std::shared_ptr<NNTensor>& tensor) in SetHasBias()
75 OH_NN_ReturnCode LSTMBuilder::SetInputSize(const std::shared_ptr<NNTensor>& tensor) in SetInputSize()
97 OH_NN_ReturnCode LSTMBuilder::SetHiddenSize(const std::shared_ptr<NNTensor>& tensor) in SetHiddenSize()
119 OH_NN_ReturnCode LSTMBuilder::SetNumLayers(const std::shared_ptr<NNTensor>& tensor) in SetNumLayers()
141 OH_NN_ReturnCode LSTMBuilder::SetNumDirections(const std::shared_ptr<NNTensor>& tensor) in SetNumDirections()
163 OH_NN_ReturnCode LSTMBuilder::SetDropout(const std::shared_ptr<NNTensor>& tensor) in SetDropout()
185 OH_NN_ReturnCode LSTMBuilder::SetZoneoutCell(const std::shared_ptr<NNTensor>& tensor) in SetZoneoutCell()
207 OH_NN_ReturnCode LSTMBuilder::SetZoneoutHidden(const std::shared_ptr<NNTensor>& tensor) in SetZoneoutHidden()
229 OH_NN_ReturnCode LSTMBuilder::SetProjSize(const std::shared_ptr<NNTensor>& tensor) in SetProjSize()
[all …]
|
| H A D | lrn_builder.cpp | 35 OH_NN_ReturnCode LRNBuilder::SetDepthRadius(const std::shared_ptr<NNTensor>& tensor) in SetDepthRadius()
57 OH_NN_ReturnCode LRNBuilder::SetBias(const std::shared_ptr<NNTensor>& tensor) in SetBias()
79 OH_NN_ReturnCode LRNBuilder::SetAlpha(const std::shared_ptr<NNTensor>& tensor) in SetAlpha()
101 OH_NN_ReturnCode LRNBuilder::SetBeta(const std::shared_ptr<NNTensor>& tensor) in SetBeta()
123 OH_NN_ReturnCode LRNBuilder::SetNormRegion(const std::shared_ptr<NNTensor>& tensor) in SetNormRegion()
179 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local
|
| H A D | resize_bilinear_builder.cpp | 33 OH_NN_ReturnCode ResizeBilinearBuilder::SetNewHeight(const std::shared_ptr<NNTensor>& tensor) in SetNewHeight()
56 OH_NN_ReturnCode ResizeBilinearBuilder::SetNewWidth(const std::shared_ptr<NNTensor>& tensor) in SetNewWidth()
79 …N_ReturnCode ResizeBilinearBuilder::SetPreserveAspectRatio(const std::shared_ptr<NNTensor>& tensor) in SetPreserveAspectRatio()
102 …turnCode ResizeBilinearBuilder::SetCoordinateTransformMode(const std::shared_ptr<NNTensor>& tensor) in SetCoordinateTransformMode()
128 OH_NN_ReturnCode ResizeBilinearBuilder::SetExcludeOutside(const std::shared_ptr<NNTensor>& tensor) in SetExcludeOutside()
177 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local
|
| H A D | strided_slice_builder.cpp | 50 OH_NN_ReturnCode StridedSliceBuilder::SetBeginMask(const std::shared_ptr<NNTensor>& tensor) in SetBeginMask()
67 OH_NN_ReturnCode StridedSliceBuilder::SetEndMask(const std::shared_ptr<NNTensor>& tensor) in SetEndMask()
84 OH_NN_ReturnCode StridedSliceBuilder::SetEllipsisMask(const std::shared_ptr<NNTensor>& tensor) in SetEllipsisMask()
101 OH_NN_ReturnCode StridedSliceBuilder::SetNewAxisMask(const std::shared_ptr<NNTensor>& tensor) in SetNewAxisMask()
118 OH_NN_ReturnCode StridedSliceBuilder::SetShrinkAxisMask(const std::shared_ptr<NNTensor>& tensor) in SetShrinkAxisMask()
164 std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local
|
| H A D | argmax_builder.cpp | 30 OH_NN_ReturnCode ArgMaxBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()
49 OH_NN_ReturnCode ArgMaxBuilder::SetTopK(const std::shared_ptr<NNTensor>& tensor) in SetTopK()
68 OH_NN_ReturnCode ArgMaxBuilder::SetKeepdims(const std::shared_ptr<NNTensor>& tensor) in SetKeepdims()
87 OH_NN_ReturnCode ArgMaxBuilder::SetOutMaxValue(const std::shared_ptr<NNTensor>& tensor) in SetOutMaxValue()
136 const std::shared_ptr<NNTensor> tensor = allTensors[i]; in Build() local
|
| H A D | pooling_builder.cpp | 59 std::shared_ptr<NNTensor> tensor = allTensors[i]; in PoolingBuild() local
95 OH_NN_ReturnCode PoolingBuilder::SetKernel(const std::shared_ptr<NNTensor>& tensor) in SetKernel()
117 OH_NN_ReturnCode PoolingBuilder::SetStrides(const std::shared_ptr<NNTensor>& tensor) in SetStrides()
139 OH_NN_ReturnCode PoolingBuilder::SetPadModeOrPaddings(const std::shared_ptr<NNTensor>& tensor) in SetPadModeOrPaddings()
183 OH_NN_ReturnCode PoolingBuilder::SetRoundMode(const std::shared_ptr<NNTensor>& tensor) in SetRoundMode()
216 OH_NN_ReturnCode PoolingBuilder::SetActivation(const std::shared_ptr<NNTensor>& tensor) in SetActivation()
247 OH_NN_ReturnCode PoolingBuilder::SetGlobal(const std::shared_ptr<NNTensor>& tensor) in SetGlobal()
|