Home
last modified time | relevance | path

Searched defs:pooling_param (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dpooling_populate_v0.cc32 auto *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in PopulatePoolingParameter() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c20 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in AvgPoolingFp16()
138 …ingC8Fp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingC8Fp16()
182 …ingC4Fp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingC4Fp16()
228 …ingC1Fp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingC1Fp16()
253 …olingFp16(const float16_t *input_ptr, float16_t *output_ptr, const PoolingParameter *pooling_param, in MaxPoolingFp16()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dpooling_grad_fp32_tests.cc34 void InitPoolingParamFP32(PoolingParameter *pooling_param) { in InitPoolingParamFP32()
61 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
116 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
190 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
392 auto pooling_param = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c21 …gPoolingInt8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, i… in AvgPoolingInt8()
82 …olingOptInt8(const int8_t *input_ptr, int8_t *output_ptr, const PoolingParameter *pooling_param, i… in AvgPoolingOptInt8()
277 void MaxPoolingInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param, i… in MaxPoolingInt8()
327 …xPoolingWithQuantInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param, in MaxPoolingWithQuantInt8()
423 void MaxPoolingOptInt8(const int8_t *input_ptr, int8_t *output_ptr, PoolingParameter *pooling_param in MaxPoolingOptInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpooling_grad.c21 …ad(const float16_t *input_ptr, float16_t *output_ptr, int count, PoolingParameter *pooling_param) { in AvgPoolingFp16Grad()
106 PoolingParameter *pooling_param) { in MaxPoolingFp16Grad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpooling_grad.c21 …Grad(const float *input_ptr, float *output_ptr, int count, const PoolingParameter *pooling_param) { in AvgPoolingGrad()
104 const PoolingParameter *pooling_param) { in MaxPoolingGrad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpooling_fp32.c22 int AvgPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in… in AvgPooling()
136 int MaxPooling(const float *input_ptr, float *output_ptr, const PoolingParameter *pooling_param, in… in MaxPooling()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc177 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulateMaxPoolGradParameter() local
216 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulateAvgPoolGradParameter() local
Dtrain_populate_parameter_v0.cc269 …PoolingParameter *pooling_param = reinterpret_cast<PoolingParameter *>(malloc(sizeof(PoolingParame… in PopulatePoolingGradParameter() local
/third_party/mindspore/third_party/proto/caffe/
Dcaffe.proto393 optional PoolingParameter pooling_param = 121; field
1343 optional PoolingParameter pooling_param = 19; field