Searched refs:pooled_w (Results 1 – 5 of 5) sorted by relevance
52 const int pooled_w = pool_info.pooled_width(); in roi_pool_layer() local90 …uto region_start_x = static_cast<int>(std::floor((static_cast<float>(px) / pooled_w) * roi_width)); in roi_pool_layer()91 …region_end_x = static_cast<int>(std::floor((static_cast<float>(px + 1) / pooled_w) * roi_width)); in roi_pool_layer()
163 const int pooled_w = _pool_info.pooled_width(); in run() local190 for(int px = 0; px < pooled_w; ++px) in run()192 …uto region_start_x = static_cast<int>(std::floor((static_cast<float>(px) / pooled_w) * roi_width)); in run()193 …region_end_x = static_cast<int>(std::floor((static_cast<float>(px + 1) / pooled_w) * roi_width)); in run()
230 const int pooled_w = pool_info.pooled_width(); in roi_align() local270 for(int px = 0; px < pooled_w; ++px) in roi_align()
228 const unsigned int pooled_w = dst->dimension(idx_width); in validate_and_configure_window() local270 dst_shape.set(0, pooled_w); in validate_and_configure_window()
803 int pooled_w = 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()822 output_shape.set(idx_width, static_cast<size_t>(pooled_w)); in compute_pool_shape()