Searched refs:pooled_h (Results 1 – 5 of 5) sorted by relevance
53 const int pooled_h = pool_info.pooled_height(); in roi_pool_layer() local92 …to region_start_y = static_cast<int>(std::floor((static_cast<float>(py) / pooled_h) * roi_height)); in roi_pool_layer()93 …egion_end_y = static_cast<int>(std::floor((static_cast<float>(py + 1) / pooled_h) * roi_height)); in roi_pool_layer()
164 const int pooled_h = _pool_info.pooled_height(); in run() local188 for(int py = 0; py < pooled_h; ++py) in run()194 …to region_start_y = static_cast<int>(std::floor((static_cast<float>(py) / pooled_h) * roi_height)); in run()195 …egion_end_y = static_cast<int>(std::floor((static_cast<float>(py + 1) / pooled_h) * roi_height)); in run()
231 const int pooled_h = pool_info.pooled_height(); in roi_align() local268 for(int py = 0; py < pooled_h; ++py) in roi_align()
229 const unsigned int pooled_h = dst->dimension(idx_height); in validate_and_configure_window() local271 dst_shape.set(1, pooled_h); in validate_and_configure_window()
804 int pooled_h = 0; in compute_pool_shape() local816 std::tie(pooled_w, pooled_h) = scaled_dimensions_signed(input_width, input_height, in compute_pool_shape()820 …ARM_COMPUTE_ERROR_ON_MSG((pooled_w < 1 || pooled_h < 1), "Calculated output dimension size is inva… in compute_pool_shape()823 output_shape.set(idx_height, static_cast<size_t>(pooled_h)); in compute_pool_shape()