Searched defs:pooled_h (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | roi_pooling.cc | 27 void ROIPooling::set_pooled_h(const int64_t pooled_h) { (void)this->AddAttr(kPooledH, api::MakeValu… in set_pooled_h()
45 void ROIPooling::Init(const int64_t pooled_h, const int64_t pooled_w, const float scale) { in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/infer/ |
| D | dpico_psroi_pool_infer.cc | 93 int pooled_h = group_size; in Infer() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_roi_align.py | 46 def __init__(self, pooled_h, pooled_w, spatial_scale, strategy=None): argument
|
| /third_party/mindspore/mindspore-src/source/third_party/proto/caffe/ |
| D | caffe.proto | 1106 optional uint32 pooled_h = 1 [default = 0]; // The pooled output height field
1639 optional uint32 pooled_h = 1 [default = 0]; // The pooled output height field
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_1.cc | 1260 PrimitivePtr MindIR_ROIPooling_CreatePrimitive(int64_t pooled_h, int64_t pooled_w, float scale) { in MindIR_ROIPooling_CreatePrimitive()
1284 void MindIR_ROIPooling_SetPooledH(PrimitivePtr *primitive, int64_t pooled_h) { in MindIR_ROIPooling_SetPooledH()
|