Searched refs:num_rois (Results 1 – 5 of 5) sorted by relevance
44 const size_t num_rois = rois.shape()[1]; in roi_pool_layer() local49 … output_shape(pool_info.pooled_width(), pool_info.pooled_height(), src.shape()[2], num_rois); in roi_pool_layer()67 for(size_t pw = 0; pw < num_rois; ++pw) in roi_pool_layer()
145 const size_t num_rois = rois.shape()[1]; in roi_align_layer() local151 … output_shape(pool_info.pooled_width(), pool_info.pooled_height(), src.shape()[2], num_rois); in roi_align_layer()159 for(size_t pw = 0; pw < num_rois; ++pw) in roi_align_layer()
156 const unsigned int num_rois = rois->info()->dimension(1); in configure() local158 window.set(Window::DimX, Window::Dimension(0, num_rois)); in configure()
66 const size_t num_rois = rois_shape.y();88 for(unsigned int pw = 0; pw < num_rois; ++pw)
67 const size_t num_rois = rois_shape.y();89 for(unsigned int pw = 0; pw < num_rois; ++pw)