| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | pooling_populate_v0.cc | 32 auto *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in PopulatePoolingParameter() local
33 if (pooling_param == nullptr) { in PopulatePoolingParameter()
37 memset(pooling_param, 0, sizeof(PoolingParameter)); in PopulatePoolingParameter()
38 pooling_param->global_ = pooling_prim->global(); in PopulatePoolingParameter()
39 pooling_param->window_w_ = pooling_prim->windowW(); in PopulatePoolingParameter()
40 pooling_param->window_h_ = pooling_prim->windowH(); in PopulatePoolingParameter()
41 pooling_param->pad_u_ = pooling_prim->padUp(); in PopulatePoolingParameter()
42 pooling_param->pad_d_ = pooling_prim->padDown(); in PopulatePoolingParameter()
43 pooling_param->pad_l_ = pooling_prim->padLeft(); in PopulatePoolingParameter()
44 pooling_param->pad_r_ = pooling_prim->padRight(); in PopulatePoolingParameter()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 21 …(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, int task_id) { in AvgPoolingInt8() argument
22 int stride_w = pooling_param->stride_w_; in AvgPoolingInt8()
23 int stride_h = pooling_param->stride_h_; in AvgPoolingInt8()
24 int pad_w = pooling_param->pad_l_; in AvgPoolingInt8()
25 int pad_h = pooling_param->pad_u_; in AvgPoolingInt8()
26 int win_w = pooling_param->window_w_; in AvgPoolingInt8()
27 int win_h = pooling_param->window_h_; in AvgPoolingInt8()
28 int channel = pooling_param->input_channel_; in AvgPoolingInt8()
29 int in_w = pooling_param->input_w_; in AvgPoolingInt8()
30 int in_h = pooling_param->input_h_; in AvgPoolingInt8()
[all …]
|
| D | pooling_int8.h | 31 …8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, int task_id);
33 …8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, int task_id);
35 void MaxPoolingInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param, i…
37 …antInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param, int task_id);
39 void MaxPoolingOptInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param…
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | pooling_grad_fp32_tests.cc | 34 void InitPoolingParamFP32(PoolingParameter *pooling_param) { in InitPoolingParamFP32() argument
35 pooling_param->input_batch_ = 1; in InitPoolingParamFP32()
36 pooling_param->input_h_ = 28; in InitPoolingParamFP32()
37 pooling_param->input_w_ = 28; in InitPoolingParamFP32()
38 pooling_param->input_channel_ = 3; in InitPoolingParamFP32()
40 pooling_param->output_batch_ = 1; in InitPoolingParamFP32()
41 pooling_param->output_h_ = 28; in InitPoolingParamFP32()
42 pooling_param->output_w_ = 28; in InitPoolingParamFP32()
43 pooling_param->output_channel_ = 32; in InitPoolingParamFP32()
45 pooling_param->window_h_ = 3; in InitPoolingParamFP32()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 20 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in AvgPoolingFp16() argument
22 int win_w = pooling_param->window_w_; in AvgPoolingFp16()
23 int win_h = pooling_param->window_h_; in AvgPoolingFp16()
24 int channel = pooling_param->input_channel_; in AvgPoolingFp16()
26 int in_w = pooling_param->input_w_; in AvgPoolingFp16()
27 int in_h = pooling_param->input_h_; in AvgPoolingFp16()
28 int output_w = pooling_param->output_w_; in AvgPoolingFp16()
29 int output_h = pooling_param->output_h_; in AvgPoolingFp16()
39 for (int batch = 0; batch < pooling_param->output_batch_; batch++) { in AvgPoolingFp16()
42 …for (int thread_id = task_id; thread_id < out_tile_count; thread_id += pooling_param->thread_num_)… in AvgPoolingFp16()
[all …]
|
| D | pooling_fp16.h | 26 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param,
29 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 22 int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in… in AvgPooling() argument
24 int win_w = pooling_param->window_w_; in AvgPooling()
25 int win_h = pooling_param->window_h_; in AvgPooling()
26 int channel = pooling_param->input_channel_; in AvgPooling()
27 int in_w = pooling_param->input_w_; in AvgPooling()
28 int in_h = pooling_param->input_h_; in AvgPooling()
29 int output_w = pooling_param->output_w_; in AvgPooling()
30 int output_h = pooling_param->output_h_; in AvgPooling()
45 for (int batch = 0; batch < pooling_param->output_batch_; batch++) { in AvgPooling()
48 …for (int thread_id = task_id; thread_id < out_tile_count; thread_id += pooling_param->thread_num_)… in AvgPooling()
[all …]
|
| D | pooling_fp32.h | 30 int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in…
32 int MaxPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pooling_grad.c | 21 …Grad(const float *input_ptr, float *output_ptr, int count, const PoolingParameter *pooling_param) { in AvgPoolingGrad() argument
22 int stride_w = pooling_param->stride_w_; in AvgPoolingGrad()
23 int stride_h = pooling_param->stride_h_; in AvgPoolingGrad()
24 int pad_w = pooling_param->pad_l_; in AvgPoolingGrad()
25 int pad_h = pooling_param->pad_u_; in AvgPoolingGrad()
26 int win_w = pooling_param->window_w_; in AvgPoolingGrad()
27 int win_h = pooling_param->window_h_; in AvgPoolingGrad()
28 int channel = pooling_param->input_channel_; in AvgPoolingGrad()
29 int in_w = pooling_param->input_w_; in AvgPoolingGrad()
30 int in_h = pooling_param->input_h_; in AvgPoolingGrad()
[all …]
|
| D | pooling_grad.h | 25 …gGrad(const float *input_ptr, float *output_ptr, int count, const PoolingParameter *pooling_param);
27 const PoolingParameter *pooling_param);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pooling_grad.c | 21 …ad(const float16_t *input_ptr, float16_t *output_ptr, int count, PoolingParameter *pooling_param) { in AvgPoolingFp16Grad() argument
22 int stride_w = pooling_param->stride_w_; in AvgPoolingFp16Grad()
23 int stride_h = pooling_param->stride_h_; in AvgPoolingFp16Grad()
24 int pad_w = pooling_param->pad_l_; in AvgPoolingFp16Grad()
25 int pad_h = pooling_param->pad_u_; in AvgPoolingFp16Grad()
26 int win_w = pooling_param->window_w_; in AvgPoolingFp16Grad()
27 int win_h = pooling_param->window_h_; in AvgPoolingFp16Grad()
28 int channel = pooling_param->input_channel_; in AvgPoolingFp16Grad()
29 int in_w = pooling_param->input_w_; in AvgPoolingFp16Grad()
30 int in_h = pooling_param->input_h_; in AvgPoolingFp16Grad()
[all …]
|
| D | pooling_grad.h | 25 …rad(const float16_t *input_ptr, float16_t *output_ptr, int count, PoolingParameter *pooling_param);
27 PoolingParameter *pooling_param);
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_populate_parameter.cc | 177 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulateMaxPoolGradParameter() local
178 if (pooling_param == nullptr) { in PopulateMaxPoolGradParameter()
182 memset(pooling_param, 0, sizeof(PoolingParameter)); in PopulateMaxPoolGradParameter()
186 pooling_param->op_parameter_.type_ = primitive->value_type(); in PopulateMaxPoolGradParameter()
188 pooling_param->global_ = false; in PopulateMaxPoolGradParameter()
189 pooling_param->window_w_ = static_cast<int>(value->kernel_size()->Get(1)); in PopulateMaxPoolGradParameter()
190 pooling_param->window_h_ = static_cast<int>(value->kernel_size()->Get(0)); in PopulateMaxPoolGradParameter()
192 pooling_param->pad_u_ = 0; in PopulateMaxPoolGradParameter()
193 pooling_param->pad_d_ = 0; in PopulateMaxPoolGradParameter()
194 pooling_param->pad_l_ = 0; in PopulateMaxPoolGradParameter()
[all …]
|
| D | train_populate_parameter_v0.cc | 269 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulatePoolingGradParameter() local
270 if (pooling_param == nullptr) { in PopulatePoolingGradParameter()
274 memset(pooling_param, 0, sizeof(PoolingParameter)); in PopulatePoolingGradParameter()
278 pooling_param->global_ = poolingGrad_prim->global(); in PopulatePoolingGradParameter()
279 pooling_param->window_w_ = poolingGrad_prim->windowW(); in PopulatePoolingGradParameter()
280 pooling_param->window_h_ = poolingGrad_prim->windowH(); in PopulatePoolingGradParameter()
282 pooling_param->pad_u_ = poolingGrad_prim->padUp(); in PopulatePoolingGradParameter()
283 pooling_param->pad_d_ = poolingGrad_prim->padDown(); in PopulatePoolingGradParameter()
284 pooling_param->pad_l_ = poolingGrad_prim->padLeft(); in PopulatePoolingGradParameter()
285 pooling_param->pad_r_ = poolingGrad_prim->padRight(); in PopulatePoolingGradParameter()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_pooling_parser.cc | 109 const caffe::PoolingParameter &poolingParam = proto.pooling_param(); in Parse()
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 393 optional PoolingParameter pooling_param = 121; field
1343 optional PoolingParameter pooling_param = 19; field
|