Home
last modified time | relevance | path

Searched refs:output_n_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Droi_pooling_fp32.h38 int output_n_; member
Droi_pooling_fp32.c29 int num_rois = param->output_n_; in ROIPooling()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Droi_pooling_fp32.cc64 param_->output_n_ = out_shape.at(0); in ReSize()