| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | pad_builder_test.cpp | 39 PadBuilder m_pad; member in OHOS::NeuralNetworkRuntime::UnitTest::PadBuilderTest
84 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
100 EXPECT_EQ(OH_NN_SUCCESS, m_pad.Build(m_params, m_inputsIndex, m_outputsIndex, m_allTensors)); in HWTEST_F()
101 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
121 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
140 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
151 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputs, m_outputs, m_allTensors); in HWTEST_F()
164 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputsIndex, m_outputs, m_allTensors); in HWTEST_F()
184 OH_NN_ReturnCode ret = m_pad.Build(m_params, m_inputsIndex, m_outputsIndex, m_allTensors); in HWTEST_F()
206 OH_NN_ReturnCode ret = m_pad in HWTEST_F()
[all...] |
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | maxpool_builder.cpp | 49 void* primitive = MindIR_MaxPoolFusion_CreatePrimitive(m_kernelSize, m_strides, m_pad, in GetPrimitive()
|
| H A D | avgpool_builder.cpp | 50 void* primitive = mindspore::lite::MindIR_AvgPoolFusion_CreatePrimitive(m_kernelSize, m_strides, m_pad, in GetPrimitive()
|
| H A D | conv2d_builder.cpp | 167 m_pad.assign(pPadList, pPadList + padListSize); in SetPad()
284 m_dilation, m_padMode, m_pad, m_group, m_inChannel, m_outChannel, m_activationType); in GetPrimitive()
|
| H A D | pooling_builder.cpp | 171 m_pad.clear(); in SetPadModeOrPaddings()
173 m_pad.emplace_back(static_cast<int64_t>(pPad[i])); in SetPadModeOrPaddings()
|
| H A D | depthwise_conv2d_native_builder.cpp | 179 m_pad.assign(pPadList, pPadList + padListSize); in SetPadModeOrPaddings()
263 m_dilation, m_padMode, m_pad, m_inChannel, m_inChannel, m_outChannel, m_activationType); in GetPrimitive()
|
| H A D | depthwise_conv2d_native_builder.h | 55 std::vector<int64_t> m_pad; member in OHOS::NeuralNetworkRuntime::Ops::DepthwiseConv2DNativeBuilder
|
| H A D | conv2d_builder.h | 59 std::vector<int64_t> m_pad; member in OHOS::NeuralNetworkRuntime::Ops::Conv2DBuilder
|
| H A D | pooling_builder.h | 51 std::vector<int64_t> m_pad; member in OHOS::NeuralNetworkRuntime::Ops::PoolingBuilder
|