Searched defs:SetPoolingParam (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ | ||
| D | max_pooling_npu.cc | 30 int MaxPoolingNPUOp::SetPoolingParam(const schema::MaxPoolFusion *pooling_prim) { in SetPoolingParam() function in mindspore::MaxPoolingNPUOp |
| D | avg_pooling_npu.cc | 38 int AvgPoolingNPUOp::SetPoolingParam(const schema::AvgPoolFusion *pooling_prim) { in SetPoolingParam() function in mindspore::AvgPoolingNPUOp |