Home
last modified time | relevance | path

Searched defs:pool_end_y (Results 1 – 3 of 3) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/pool3d/neon/
Dimpl.cpp90 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in max_poolingMxNxD_fp_neon_ndhwc() local
202 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in avg_poolingMxNxD_fp_neon_ndhwc() local
329 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in l2_poolingMxNxD_fp_neon_ndhwc() local
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/
Dfp32.cpp186 const int pool_end_y = std::min(pool_size_y, window_src.z().end() + pool_limit_y); in poolingMxN_fp32_neon_nhwc() local
Dfp16.cpp192 const int pool_end_y = std::min(pool_size_y, window_src.z().end() + pool_limit_y); in poolingMxN_fp16_neon_nhwc() local