Searched refs:pool_end_z_t (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/pool3d/neon/ |
| D | impl.cpp | 86 const int pool_end_z_t = std::min(input_dim_d + pool_pad_front - in_idx_depth, pool_size_z); in max_poolingMxNxD_fp_neon_ndhwc() local
91 const int pool_end_z = std::min(pool_end_z_t, input_dim_d - in_idx_depth); in max_poolingMxNxD_fp_neon_ndhwc()
198 const int pool_end_z_t = std::min(input_dim_d + pool_pad_front - in_idx_depth, pool_size_z); in avg_poolingMxNxD_fp_neon_ndhwc() local
203 const int pool_end_z = std::min(pool_end_z_t, input_dim_d - in_idx_depth); in avg_poolingMxNxD_fp_neon_ndhwc()
325 const int pool_end_z_t = std::min(input_dim_d + pool_pad_front - in_idx_depth, pool_size_z); in l2_poolingMxNxD_fp_neon_ndhwc() local
330 const int pool_end_z = std::min(pool_end_z_t, input_dim_d - in_idx_depth); in l2_poolingMxNxD_fp_neon_ndhwc()
|
| D | quantized.h | 107 const int pool_end_z_t = std::min(input_dim_d + pool_pad_front - in_idx_depth, pool_size_z); in avg_poolingMxNxD_q8_neon_ndhwc()
112 const int pool_end_z = std::min(pool_end_z_t, input_dim_d - in_idx_depth); in avg_poolingMxNxD_q8_neon_ndhwc()
286 const int pool_end_z_t = std::min(input_dim_d + pool_pad_front - in_idx_depth, pool_size_z); in max_poolingMxNxD_q8_neon_ndhwc()
291 const int pool_end_z = std::min(pool_end_z_t, input_dim_d - in_idx_depth); in max_poolingMxNxD_q8_neon_ndhwc()
|