| /external/ComputeLibrary/src/core/helpers/ |
| D | PoolingHelpers.h | 37 …const int upper_bound_h, const int upper_bound_d, const int pad_x, const int pad_y, const int pad_… in calculate_avg_scale_pool3d() argument
45 const int end_y = std::min(start_y + pool_size_y, upper_bound_h); in calculate_avg_scale_pool3d()
56 …id, const int pool_size_x, const int pool_size_y, const int upper_bound_w, const int upper_bound_h, in calculate_avg_scale_pool2d() argument
66 const int end_y = std::min(start_y + pool_size_y, upper_bound_h); in calculate_avg_scale_pool2d()
|
| /external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/ |
| D | all.cpp | 95 …const int upper_bound_h = src_h + (pool_info.exclude_padding ? 0 : pool_pad_bott… in pooling3_fp16_neon_nchw() local
127 …e_padding, DataLayout::NCHW, id, pool_size, pool_size, upper_bound_w, upper_bound_h, pool_pad_left… in pooling3_fp16_neon_nchw()
261 const int upper_bound_h = src_h + (pool_info.exclude_padding ? 0 : pool_pad_bottom); in pooling2_fp16_neon_nchw() local
291 …e_padding, DataLayout::NCHW, id, pool_size, pool_size, upper_bound_w, upper_bound_h, pool_pad_left… in pooling2_fp16_neon_nchw()
335 const int upper_bound_h = src_h + (pool_info.exclude_padding ? 0 : pool_pad_bottom); in poolingMxN_fp16_neon_nchw() local
346 …dding, DataLayout::NCHW, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_fp16_neon_nchw()
423 const int upper_bound_h = src_h + (pool_info.exclude_padding ? 0 : pool_pad_bottom); in poolingMxN_fp32_neon_nchw() local
433 …info.exclude_padding, DataLayout::NCHW, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, in poolingMxN_fp32_neon_nchw()
512 const int upper_bound_h = src_h + (pool_info.exclude_padding ? 0 : pool_pad_bottom); in pooling2_fp32_neon_nchw() local
541 …e_padding, DataLayout::NCHW, id, pool_size, pool_size, upper_bound_w, upper_bound_h, pool_pad_left… in pooling2_fp32_neon_nchw()
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/pool2d/neon/ |
| D | quantized.h | 74 …const int upper_bound_h = src->info()->dimension(2) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in poolingMxN_q8_neon_nhwc() local
112 …dding, DataLayout::NHWC, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_q8_neon_nhwc()
214 …dding, DataLayout::NHWC, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_q8_neon_nhwc()
280 … const int pool_size, const int upper_bound_w, const int upper_bound_h, in scale_vector_q16x8() argument
285 const int end_y = std::min(start_y + pool_size, upper_bound_h); in scale_vector_q16x8()
409 …const int upper_bound_h = src->info()->dimension(1) + (pool_info.exclud… in pooling2_quantized_neon_nchw() local
462 pool_size, upper_bound_w, upper_bound_h, in pooling2_quantized_neon_nchw()
485 pool_size, upper_bound_w, upper_bound_h, in pooling2_quantized_neon_nchw()
545 …const int upper_bound_h = src->info()->dimension(1) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in pooling3_quantized_neon_nchw() local
632 pool_size, upper_bound_w, upper_bound_h, in pooling3_quantized_neon_nchw()
[all …]
|
| D | fp32.cpp | 174 …const int upper_bound_h = src->info()->dimension(2) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in poolingMxN_fp32_neon_nhwc() local
196 …dding, DataLayout::NHWC, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_fp32_neon_nhwc()
261 …dding, DataLayout::NHWC, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_fp32_neon_nhwc()
|
| D | fp16.cpp | 180 …const int upper_bound_h = src->info()->dimension(2) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in poolingMxN_fp16_neon_nhwc() local
202 …dding, DataLayout::NHWC, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_fp16_neon_nhwc()
263 …dding, DataLayout::NHWC, id, pool_size_x, pool_size_y, upper_bound_w, upper_bound_h, pool_pad_left… in poolingMxN_fp16_neon_nhwc()
|
| /external/ComputeLibrary/src/cpu/kernels/pool3d/neon/ |
| D | impl.cpp | 168 …const int upper_bound_h = src->info()->dimension(2) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in avg_poolingMxNxD_fp_neon_ndhwc() local
208 …de_padding, id, pool_size_x, pool_size_y, pool_size_z, upper_bound_w, upper_bound_h, upper_bound_d… in avg_poolingMxNxD_fp_neon_ndhwc()
295 …const int upper_bound_h = src->info()->dimension(2) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in l2_poolingMxNxD_fp_neon_ndhwc() local
335 …de_padding, id, pool_size_x, pool_size_y, pool_size_z, upper_bound_w, upper_bound_h, upper_bound_d… in l2_poolingMxNxD_fp_neon_ndhwc()
|
| D | quantized.h | 65 …const int upper_bound_h = src->info()->dimension(2) + (pool_info.exclude_padding ? 0 : pool_pad_bo… in avg_poolingMxNxD_q8_neon_ndhwc() local
115 …de_padding, id, pool_size_x, pool_size_y, pool_size_z, upper_bound_w, upper_bound_h, upper_bound_d… in avg_poolingMxNxD_q8_neon_ndhwc()
|
| /external/ComputeLibrary/src/core/CL/cl_kernels/nchw/ |
| D | pooling_layer.cl | 52 …ale(const int pool_size_x, const int pool_size_y, const int upper_bound_w, const int upper_bound_h,
58 const int end_y = min(start_y + pool_size_y, upper_bound_h);
|
| /external/ComputeLibrary/cl_kernels/nchw/ |
| D | pooling_layer.clembed | 1273 …ale(const int pool_size_x, const int pool_size_y, const int upper_bound_w, const int upper_bound_h,
1279 const int end_y = min(start_y + pool_size_y, upper_bound_h);
|