| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | avgpool_padmod_test.cpp | 88 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in SetParams()
240 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
268 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
295 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
321 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
347 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
374 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
398 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
422 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
445 SetKernelSize(OH_NN_INT6 in HWTEST_F()
[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()
235 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
261 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
286 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
313 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
339 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
366 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
390 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
414 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
437 SetKernelSize(OH_NN_INT6 in HWTEST_F()
[all...] |
| H A D | avgpool_pad_test.cpp | 90 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in SetPadParams()
240 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
267 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
293 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
319 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
345 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
372 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
419 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
441 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_AVG_POOL_KERNEL_SIZE); in HWTEST_F()
463 SetKernelSize(OH_NN_INT6 in HWTEST_F()
[all...] |
| H A D | maxpool_pad_test.cpp | 91 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in SetPadParam()
241 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
267 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
294 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
320 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
346 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
373 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
420 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
442 SetKernelSize(OH_NN_INT64, m_kenelsize_dim, nullptr, OH_NN_MAX_POOL_KERNEL_SIZE); in HWTEST_F()
464 SetKernelSize(OH_NN_INT6 in HWTEST_F()
[all...] |
| H A D | ops_test.h | 38 void SetKernelSize(OH_NN_DataType dataType,
|
| H A D | ops_test.cpp | 45 void OpsTest::SetKernelSize(OH_NN_DataType dataType, in SetKernelSize() function in OHOS::NeuralNetworkRuntime::UnitTest::OpsTest
|
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | conv2d_builder.cpp | 75 void Conv2DBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() function in OHOS::NeuralNetworkRuntime::Ops::Conv2DBuilder
251 SetKernelSize(inputsIndex, allTensors); in Build()
|
| H A D | depthwise_conv2d_native_builder.cpp | 87 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() function in OHOS::NeuralNetworkRuntime::Ops::DepthwiseConv2DNativeBuilder
93 LOGE("[DepthwiseConv2DNative] SetKernelSize failed, invalid rank of shape of weight, should be 4 dimensions."); in SetKernelSize()
227 ret = SetKernelSize(inputsIndex, allTensors); in Build()
229 LOGE("[DepthwiseConv2DNative] Build failed, SetKernelSize failed."); in Build()
|
| H A D | depthwise_conv2d_native_builder.h | 44 OH_NN_ReturnCode SetKernelSize(const std::vector<uint32_t>& inputsIndex,
|
| H A D | conv2d_transpose_builder.h | 43 void SetKernelSize(const std::vector<uint32_t>& inputsIndex,
|
| H A D | conv2d_builder.h | 45 void SetKernelSize(const std::vector<uint32_t>& inputsIndex,
|
| H A D | conv2d_transpose_builder.cpp | 69 void Conv2DTransposeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() function in OHOS::NeuralNetworkRuntime::Ops::Conv2DTransposeBuilder
261 SetKernelSize(inputsIndex, allTensors); in Build()
|