/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | lstm_test.cpp | 241 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 266 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 296 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 324 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 378 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 408 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 438 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 468 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 498 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); 528 SaveProjSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_PROJ_SIZE); [all …]
|
H A D | detection_post_process_test.cpp | 198 SetInputSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_INPUT_SIZE); 222 SetInputSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_INPUT_SIZE); 252 SetInputSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_INPUT_SIZE); 279 SetInputSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_DETECTION_POST_PROCESS_INPUT_SIZE); 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); 724 SetMaxDetections(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE); 750 SetDetectionsPerClass(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE); 802 SetNumClasses(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE); 870 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, [all …]
|
H A D | conv2d_tranpose_padmode_test.cpp | 78 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_STRIDES); in SetParam() 82 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); in SetParam() 240 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 269 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 296 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 325 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 374 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 431 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 487 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 517 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); [all …]
|
H A D | conv2d_pad_test.cpp | 68 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); in SetPadParam() 225 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 251 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 297 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 323 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 349 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 376 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 405 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 429 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); 472 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_PAD); [all …]
|
H A D | quant_dtype_cast_builder_test.cpp | 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); 132 SaveDstTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_DST_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 D | range_test.cpp | 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); 151 SaveStart(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RANGE_START); [all …]
|
H A D | strided_slice_builder_test.cpp | 125 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); in InitParams() 265 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 307 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 328 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 349 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 376 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 425 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 452 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 479 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); 505 SaveEndMaskTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_END_MASK); [all …]
|
H A D | conv2d_transpose_pad_test.cpp | 84 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); in SetPadParam() 86 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); in SetPadParam() 245 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); 247 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 271 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); 273 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_GROUP); 315 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); 341 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); 366 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); 392 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_CONV2D_TRANSPOSE_PAD); [all …]
|
H A D | split_builder_test.cpp | 105 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 124 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 145 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 166 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 240 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 260 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 323 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); 325 SaveSizeSplitsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS); 343 SaveAxisTensor(OH_NN_INT64, axisDim, nullptr, OH_NN_SPLIT_AXIS); 363 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); [all …]
|
H A D | argmax_test.cpp | 106 SetArgmaxAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_AXIS); 108 SetArgmaxTopK(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_TOP_K); 128 SetArgmaxAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_AXIS); 130 SetArgmaxTopK(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_TOP_K); 152 SetArgmaxAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_AXIS); 154 SetArgmaxTopK(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_TOP_K); 175 SetArgmaxAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_AXIS); 177 SetArgmaxTopK(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_TOP_K); 198 SetArgmaxAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_AXIS); 221 SetArgmaxAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_ARG_MAX_AXIS); [all …]
|
H A D | resize_bilinear_builder_test.cpp | 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); 356 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH); 409 SaveWidthTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_RESIZE_BILINEAR_NEW_WIDTH); 469 outsideTensor = TransToNNTensor(OH_NN_INT64, 494 SaveOutsideTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_SRC_T); [all …]
|
H A D | avgpool_pad_test.cpp | 92 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); in SetPadParams() 223 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 249 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 295 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 321 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 347 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 374 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 401 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 423 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); 465 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_AVG_POOL_PAD); [all …]
|
H A D | maxpool_pad_test.cpp | 93 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); in SetPadParam() 222 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 249 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 296 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 322 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 348 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 375 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 402 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 424 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); 466 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_MAX_POOL_PAD); [all …]
|
H A D | depthwise_conv2d_native_pad_test.cpp | 67 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_STRIDES); in SetPadParam() 69 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); in SetPadParam() 227 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 252 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 298 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 324 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 352 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 378 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 505 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); 528 SetPad(OH_NN_INT64, m_pad_dim, nullptr, OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD); [all …]
|
H A D | layernorm_builder_test.cpp | 109 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 111 SaveParamAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_PARAM_AXIS); 126 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 128 SaveParamAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_PARAM_AXIS); 148 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 168 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 259 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 284 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 307 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); 332 SaveNormAixsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LAYER_NORM_BEGIN_NORM_AXIS); [all …]
|
H A D | batch_to_space_nd_test.cpp | 85 SetBlockSize(OH_NN_INT64, m_block_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_BLOCKSIZE); 86 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 102 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 123 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 143 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 163 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 183 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 229 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 270 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); 304 SetCrops(OH_NN_INT64, m_crops_dim, nullptr, OH_NN_BATCH_TO_SPACE_ND_CROPS); [all …]
|
H A D | crop_test.cpp | 94 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); 95 SaveOffset(OH_NN_INT64, m_offsetDim, nullptr, OH_NN_CROP_OFFSET); 110 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); 111 SaveOffset(OH_NN_INT64, m_offsetDim, nullptr, OH_NN_CROP_OFFSET); 134 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); 135 SaveOffset(OH_NN_INT64, m_offsetDim, nullptr, OH_NN_CROP_OFFSET); 153 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); 215 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); 253 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); 289 SaveAxis(OH_NN_INT64, m_axisDim, nullptr, OH_NN_CROP_AXIS); [all …]
|
H A D | spacetobatchnd_builder_test.cpp | 69 SaveBlockShapeTensor(OH_NN_INT64, shapeDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_BLOCK_SHAPE); in InitTensor() 70 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); in InitTensor() 124 SaveBlockShapeTensor(OH_NN_INT64, shapeDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_BLOCK_SHAPE); 125 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); 145 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); 166 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); 186 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); 206 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); 227 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); 320 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SPACE_TO_BATCH_ND_PADDINGS); [all …]
|
H A D | conv2d_padmode_test.cpp | 65 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_CONV2D_STRIDES); in SetParam() 68 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); in SetParam() 225 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); 242 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_CONV2D_STRIDES); 252 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); 282 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); 329 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); 382 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); 411 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); 443 SetGroup(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONV2D_GROUP); [all …]
|
H A D | avgpool_padmod_test.cpp | 88 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in SetParams() 89 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); in SetParams() 222 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 269 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 296 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 322 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 348 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 375 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 399 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); 423 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_AVG_POOL_STRIDE); [all …]
|
H A D | maxpool_padmode_test.cpp | 88 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in SetParam() 89 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); in SetParam() 216 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 262 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 287 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 314 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 340 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 367 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 391 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); 415 SetStride(OH_NN_INT64, m_stride_dim, nullptr, OH_NN_MAX_POOL_STRIDE); [all …]
|
H A D | lrn_test.cpp | 117 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 136 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 160 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 182 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 250 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 274 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 298 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 322 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 364 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); 383 SaveDepthRadius(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LRN_DEPTH_RADIUS); [all …]
|
H A D | concat_two_inputs_test.cpp | 71 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 86 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 106 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 125 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 144 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 163 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 200 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT… 220 …std::shared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT… 237 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 258 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS);
|
H A D | stack_builder_test.cpp | 76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 164 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 200 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 219 …std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_… 239 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); 257 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS); 287 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
|
H A D | l2_normalize_test.cpp | 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); 339 SetAxis(OH_NN_INT64, m_paramDim, nullptr, OH_NN_L2_NORMALIZE_AXIS); [all …]
|