| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | pooling_grad_fp32_tests.cc | 34 void InitPoolingArgsFp32(PoolingComputeParam *pooling_param) { in InitPoolingArgsFp32()
47 void InitPoolingParamFP32(PoolingParameter *pooling_param) { in InitPoolingParamFP32()
63 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
122 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
199 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
404 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | pooling_fp32.c | 23 int AvgPoolingBatch(const float *src_b_ptr, float *dst_b_ptr, const PoolingParameter *pooling_param, in AvgPoolingBatch()
81 int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in AvgPooling()
101 …mNC4HW4ToNHWCLessC(const float *src_b_ptr, float *dst_b_ptr, const PoolingParameter *pooling_param, in AvgPoolingFromNC4HW4ToNHWCLessC()
185 …mNC4HW4ToNHWCBatch(const float *src_b_ptr, float *dst_b_ptr, const PoolingParameter *pooling_param, in AvgPoolingFromNC4HW4ToNHWCBatch()
313 …gFromNC4HW4ToNHWC(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in AvgPoolingFromNC4HW4ToNHWC()
338 int MaxPoolingBatch(const float *src_b_ptr, float *dst_b_ptr, const PoolingParameter *pooling_param, in MaxPoolingBatch()
390 int MaxPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in MaxPooling()
410 …mNC4HW4ToNHWCLessC(const float *src_b_ptr, float *dst_b_ptr, const PoolingParameter *pooling_param, in MaxPoolingFromNC4HW4ToNHWCLessC()
491 …mNC4HW4ToNHWCBatch(const float *src_b_ptr, float *dst_b_ptr, const PoolingParameter *pooling_param, in MaxPoolingFromNC4HW4ToNHWCBatch()
612 …gFromNC4HW4ToNHWC(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingFromNC4HW4ToNHWC()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/ |
| D | avg_pooling_coreml.cc | 30 auto pooling_param = op_->mutable_pooling(); in BuildLayer() local
|
| D | max_pooling_coreml.cc | 30 auto pooling_param = op_->mutable_pooling(); in BuildLayer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | pooling_int8.c | 21 … AvgPoolingInt8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, in AvgPoolingInt8()
83 …gPoolingOptInt8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, in AvgPoolingOptInt8()
279 … MaxPoolingInt8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingInt8()
330 …xPoolingWithQuantInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param, in MaxPoolingWithQuantInt8()
426 void MaxPoolingOptInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param, in MaxPoolingOptInt8()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/ |
| D | pooling_grad.c | 22 …Grad(const float16_t *input_ptr, float16_t *output_ptr, int count, PoolingParameter *pooling_param, in AvgPoolingFp16Grad()
108 PoolingParameter *pooling_param, const PoolingComputeParam *pooling_args) { in MaxPoolingFp16Grad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | pooling_grad.c | 22 …ngGrad(const float *input_ptr, float *output_ptr, int count, const PoolingParameter *pooling_param, in AvgPoolingGrad()
106 … const PoolingParameter *pooling_param, const PoolingComputeParam *pooling_args) { in MaxPoolingGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/ |
| D | train_populate_parameter.cc | 196 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulateMaxPoolGradParameter() local
235 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulateAvgPoolGradParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | pooling_fp16.c | 21 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in AvgPoolingFp16()
257 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingFp16()
|
| /third_party/mindspore/mindspore-src/source/third_party/proto/caffe/ |
| D | caffe.proto | 393 optional PoolingParameter pooling_param = 121; field
1343 optional PoolingParameter pooling_param = 19; field
|