| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | pooling_builder.h | 47 OH_NN_ReturnCode SetGlobal(const std::shared_ptr<NNTensor>& tensor);
65 {OH_NN_MAX_POOL_GLOBAL, &PoolingBuilder::SetGlobal},
73 {OH_NN_AVG_POOL_GLOBAL, &PoolingBuilder::SetGlobal}
|
| H A D | pooling_builder.cpp | 247 OH_NN_ReturnCode PoolingBuilder::SetGlobal(const std::shared_ptr<NNTensor>& tensor) in SetGlobal() function in OHOS::NeuralNetworkRuntime::Ops::PoolingBuilder
|
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | avgpool_pad_test.cpp | 36 void SetGlobal(OH_NN_DataType dataType,
67 void AvgPoolPadBuilderTest::SetGlobal(OH_NN_DataType dataType, in SetGlobal() function in OHOS::NeuralNetworkRuntime::UnitTest::AvgPoolPadBuilderTest
95 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in SetPadParams()
226 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
252 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
278 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
304 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
330 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
404 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
426 SetGlobal(OH_NN_BOO in HWTEST_F()
[all...] |
| H A D | maxpool_pad_test.cpp | 36 void SetGlobal(OH_NN_DataType dataType,
67 void MaxPoolPadBuilderTest::SetGlobal(OH_NN_DataType dataType, in SetGlobal() function in OHOS::NeuralNetworkRuntime::UnitTest::MaxPoolPadBuilderTest
96 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in SetPadParam()
225 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
252 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
278 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
305 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
331 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
405 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
427 SetGlobal(OH_NN_BOO in HWTEST_F()
[all...] |
| H A D | avgpool_padmod_test.cpp | 36 void SetGlobal(OH_NN_DataType dataType,
66 void AvgPoolBuilderTest::SetGlobal(OH_NN_DataType dataType, in SetGlobal() function in OHOS::NeuralNetworkRuntime::UnitTest::AvgPoolBuilderTest
93 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in SetParams()
226 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
253 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
279 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
306 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
332 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_AVG_POOL_GLOBAL); in HWTEST_F()
|
| H A D | maxpool_padmode_test.cpp | 36 void SetGlobal(OH_NN_DataType dataType,
66 void MaxPoolBuilderTest::SetGlobal(OH_NN_DataType dataType, in SetGlobal() function in OHOS::NeuralNetworkRuntime::UnitTest::MaxPoolBuilderTest
93 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in SetParam()
220 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
246 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
271 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
298 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
324 SetGlobal(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_MAX_POOL_GLOBAL); in HWTEST_F()
|