| /external/ComputeLibrary/src/core/helpers/ |
| D | PoolingHelpers.h | 36 …vg_scale_pool3d(bool exclude_padding, const Coordinates &id, const int pool_size_x, const int pool… in calculate_avg_scale_pool3d() argument
44 const int end_x = std::min(start_x + pool_size_x, upper_bound_w); in calculate_avg_scale_pool3d()
56 …lude_padding, DataLayout data_layout, const Coordinates &id, const int pool_size_x, const int pool… in calculate_avg_scale_pool2d() argument
65 const int end_x = std::min(start_x + pool_size_x, upper_bound_w); in calculate_avg_scale_pool2d()
|
| /external/ComputeLibrary/src/cpu/kernels/pool3d/neon/ |
| D | impl.cpp | 51 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_info.… in max_poolingMxNxD_fp_neon_ndhwc() local
81 const int pool_end_x_t = std::min(input_dim_w + pool_pad_left - in_idx_width, pool_size_x); in max_poolingMxNxD_fp_neon_ndhwc()
156 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_info.… in avg_poolingMxNxD_fp_neon_ndhwc() local
193 const int pool_end_x_t = std::min(input_dim_w + pool_pad_left - in_idx_width, pool_size_x); in avg_poolingMxNxD_fp_neon_ndhwc()
208 …const float scale = calculate_avg_scale_pool3d(pool_info.exclude_padding, id, pool_size_x, pool_si… in avg_poolingMxNxD_fp_neon_ndhwc()
283 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_info.… in l2_poolingMxNxD_fp_neon_ndhwc() local
320 const int pool_end_x_t = std::min(input_dim_w + pool_pad_left - in_idx_width, pool_size_x); in l2_poolingMxNxD_fp_neon_ndhwc()
335 …const float scale = calculate_avg_scale_pool3d(pool_info.exclude_padding, id, pool_size_x, pool_si… in l2_poolingMxNxD_fp_neon_ndhwc()
|
| D | quantized.h | 53 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_info.… in avg_poolingMxNxD_q8_neon_ndhwc() local
102 const int pool_end_x_t = std::min(input_dim_w + pool_pad_left - in_idx_width, pool_size_x); in avg_poolingMxNxD_q8_neon_ndhwc()
115 …const float scale = calculate_avg_scale_pool3d(pool_info.exclude_padding, id, pool_size_x, pool_si… in avg_poolingMxNxD_q8_neon_ndhwc()
241 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_info.… in max_poolingMxNxD_q8_neon_ndhwc() local
281 const int pool_end_x_t = std::min(input_dim_w + pool_pad_left - in_idx_width, pool_size_x); in max_poolingMxNxD_q8_neon_ndhwc()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | PoolingLayerNode.cpp | 55 const unsigned int pool_size_x = info.is_global_pooling ? input_width : info.pool_size.width; in compute_output_descriptor() local
58 …d_width, pooled_height) = scaled_dimensions(input_width, input_height, pool_size_x, pool_size_y, i… in compute_output_descriptor()
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuPool2dKernel.cpp | 207 int pool_size_x, int pool_size_y) in validate_and_configure_window() argument
227 const bool is_square = pool_size_x == pool_size_y; in validate_and_configure_window()
240 switch(pool_size_x) in validate_and_configure_window()
339 …unsigned int pool_size_x = is_global_pooling ? src->dimension(idx_width) : pool_info.pool_size.wid… in validate() local
342 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(src, dst, pool_info, indices, Size2D(pool_size_x, p… in validate()
345 pool_size_x, pool_size_y) in validate()
|
| D | CpuPool3dKernel.cpp | 85 …const unsigned int pool_size_x = is_global_pooling ? src->dimension(idx_width) : pool_info.p… in validate_arguments() local
93 ARM_COMPUTE_RETURN_ERROR_ON((pool_size_x == 0) || (pool_size_y == 0) || (pool_size_z == 0)); in validate_arguments()
103 … pool_size_x, pool_size_y, pool_size_z, pool_info); in validate_arguments()
|
| D | CpuMaxUnpoolingLayerKernel.cpp | 84 const int pool_size_x = pool_info.pool_size.width; in validate_arguments() local
86 const Size2D pool_size(pool_size_x, pool_size_y); in validate_arguments()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClPool3dKernel.cpp | 62 …const unsigned int pool_size_x = is_global_pooling ? src->dimension(idx_width) : pool_info.p… in validate_arguments() local
74 … src->tensor_shape()[idx_depth], pool_size_x, pool_size_y, in validate_arguments()
119 …const int pool_size_x = pool_info.is_global_pooling ? src->dimension(idx_width) : pool… in configure() local
142 build_opts.add_option("-DPOOL_SIZE_X=" + support::cpp11::to_string(pool_size_x)); in configure()
|
| D | ClPool2dKernel.cpp | 56 …unsigned int pool_size_x = is_global_pooling ? src->dimension(idx_width) : pool_info.pool_si… in validate_arguments() local
64 … pool_size_x, pool_size_y, pool_info.pad_stride_info); in validate_arguments()
127 …const int pool_size_x = pool_info.is_global_pooling ? src->dimension(idx_width) : po… in configure() local
145 build_opts.add_option("-DPOOL_SIZE_X=" + support::cpp11::to_string(pool_size_x)); in configure()
|
| /external/ComputeLibrary/src/cpu/kernels/pool2d/neon/ |
| D | fp32.cpp | 164 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_i… in poolingMxN_fp32_neon_nhwc() local
188 const int pool_end_x = std::min(pool_size_x, window_src.y().end() + pool_limit_x); in poolingMxN_fp32_neon_nhwc()
196 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NHWC, id, pool_size_x, pool_size_y, u… in poolingMxN_fp32_neon_nhwc()
261 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NHWC, id, pool_size_x, pool_size_y, u… in poolingMxN_fp32_neon_nhwc()
|
| D | fp16.cpp | 170 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_i… in poolingMxN_fp16_neon_nhwc() local
194 const int pool_end_x = std::min(pool_size_x, window_src.y().end() + pool_limit_x); in poolingMxN_fp16_neon_nhwc()
202 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NHWC, id, pool_size_x, pool_size_y, u… in poolingMxN_fp16_neon_nhwc()
263 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NHWC, id, pool_size_x, pool_size_y, u… in poolingMxN_fp16_neon_nhwc()
|
| D | quantized.h | 63 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().y() : pool_i… in poolingMxN_q8_neon_nhwc() local
99 const int pool_end_x = std::min(pool_size_x, window_src.y().end() + pool_limit_x); in poolingMxN_q8_neon_nhwc()
112 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NHWC, id, pool_size_x, pool_size_y, u… in poolingMxN_q8_neon_nhwc()
214 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NHWC, id, pool_size_x, pool_size_y, u… in poolingMxN_q8_neon_nhwc()
700 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().x() : pool_i… in poolingMxN_quantized_neon_nchw() local
729 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NCHW, id, pool_size_x, pool_size_y, u… in poolingMxN_quantized_neon_nchw()
735 for(int x = 0; x < pool_size_x; ++x) in poolingMxN_quantized_neon_nchw()
752 for(int x = 0; x < pool_size_x; ++x) in poolingMxN_quantized_neon_nchw()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | PoolingLayer.cpp | 50 const int pool_size_x = info.is_global_pooling ? src.shape().x() : info.pool_size.width; in pooling_layer_internal() local
86 int wend = std::min(wstart + pool_size_x, w_src); in pooling_layer_internal()
133 int wend = std::min(wstart + pool_size_x, w_src + pad_right); in pooling_layer_internal()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLMaxUnpoolingLayerKernel.cpp | 58 const int pool_size_x = pool_info.pool_size.width; in validate_arguments() local
60 const Size2D pool_size(pool_size_x, pool_size_y); in validate_arguments()
|
| /external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/ |
| D | all.cpp | 323 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().x() : pool_i… in poolingMxN_fp16_neon_nchw() local
346 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NCHW, id, pool_size_x, pool_size_y, u… in poolingMxN_fp16_neon_nchw()
352 for(int x = 0; x < pool_size_x; ++x) in poolingMxN_fp16_neon_nchw()
379 for(int x = 0; x < pool_size_x; ++x) in poolingMxN_fp16_neon_nchw()
411 …const int pool_size_x = pool_info.is_global_pooling ? src->info()->tensor_shape().x() : pool_i… in poolingMxN_fp32_neon_nchw() local
433 …late_avg_scale_pool2d(pool_info.exclude_padding, DataLayout::NCHW, id, pool_size_x, pool_size_y, u… in poolingMxN_fp32_neon_nchw()
439 for(int x = 0; x < pool_size_x; ++x) in poolingMxN_fp32_neon_nchw()
466 for(int x = 0; x < pool_size_x; ++x) in poolingMxN_fp32_neon_nchw()
|
| /external/ComputeLibrary/src/core/CL/cl_kernels/nchw/ |
| D | pooling_layer.cl | 52 ACC_DATA_TYPE calculate_avg_scale(const int pool_size_x, const int pool_size_y, const int upper_bou…
57 const int end_x = min(start_x + pool_size_x, upper_bound_w);
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 813 …const int pool_size_x = is_global_pooling ? output_shape[idx_width] : pool_info.pool_size.w… in compute_pool_shape() local
817 pool_size_x, pool_size_y, in compute_pool_shape()
|
| /external/ComputeLibrary/cl_kernels/nchw/ |
| D | pooling_layer.clembed | 1273 ACC_DATA_TYPE calculate_avg_scale(const int pool_size_x, const int pool_size_y, const int upper_bou…
1278 const int end_x = min(start_x + pool_size_x, upper_bound_w);
|