Home
last modified time | relevance | path

Searched refs:pooledW_ (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Droi_pooling_fp32.h24 int pooledW_; member
Droi_pooling_fp32.c42 int pooled_width = param->pooledW_; in ROIPooling()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Droi_pooling_populate.cc40 param->pooledW_ = value->pooled_w(); in PopulateROIPoolingParameter()
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Droi_pooling_populate_v0.cc40 roi_pooling_param->pooledW_ = roi_pooling_prim->pooledW(); // note: origin is pooledH in PopulateROIPoolingParameter()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Droi_pooling_infer.c45 output->shape_[2] = param->pooledW_; in ROIPoolingInferShape()
/third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/
Droi_pooling_infer_test.cc42 parameter->pooledW_ = 3; in TEST_F()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Droi_pooling_fp32_tests.cc53 param->pooledW_ = 2; in TEST_F()