Home
last modified time | relevance | path

Searched refs:input_dim_h (Results 1 – 6 of 6) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/pool3d/neon/
Dimpl.cpp60 const int input_dim_h = src->info()->dimension(2); in max_poolingMxNxD_fp_neon_ndhwc() local
83 const int pool_end_y_t = std::min(input_dim_h + pool_pad_top - in_idx_height, pool_size_y); in max_poolingMxNxD_fp_neon_ndhwc()
90 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in max_poolingMxNxD_fp_neon_ndhwc()
172 const int input_dim_h = src->info()->dimension(2); in avg_poolingMxNxD_fp_neon_ndhwc() local
195 const int pool_end_y_t = std::min(input_dim_h + pool_pad_top - in_idx_height, pool_size_y); in avg_poolingMxNxD_fp_neon_ndhwc()
202 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in avg_poolingMxNxD_fp_neon_ndhwc()
299 const int input_dim_h = src->info()->dimension(2); in l2_poolingMxNxD_fp_neon_ndhwc() local
322 const int pool_end_y_t = std::min(input_dim_h + pool_pad_top - in_idx_height, pool_size_y); in l2_poolingMxNxD_fp_neon_ndhwc()
329 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in l2_poolingMxNxD_fp_neon_ndhwc()
Dquantized.h70 const int input_dim_h = src->info()->dimension(2); in avg_poolingMxNxD_q8_neon_ndhwc() local
104 const int pool_end_y_t = std::min(input_dim_h + pool_pad_top - in_idx_height, pool_size_y); in avg_poolingMxNxD_q8_neon_ndhwc()
111 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in avg_poolingMxNxD_q8_neon_ndhwc()
251 const int input_dim_h = src->info()->dimension(2); in max_poolingMxNxD_q8_neon_ndhwc() local
283 const int pool_end_y_t = std::min(input_dim_h + pool_pad_top - in_idx_height, pool_size_y); in max_poolingMxNxD_q8_neon_ndhwc()
290 const int pool_end_y = std::min(pool_end_y_t, input_dim_h - in_idx_height); in max_poolingMxNxD_q8_neon_ndhwc()
/external/ComputeLibrary/src/cpu/kernels/directconv2d/nhwc/neon/
Dimpl.cpp70 const int input_dim_h = src->info()->dimension(2); in convolve_nhwc() local
135 const int in_h_end = std::min(in_h_end_t, input_dim_h); in convolve_nhwc()
197 const int in_h_end = std::min(in_h_end_t, input_dim_h); in convolve_nhwc()
/external/ComputeLibrary/src/cpu/kernels/directconv2d/nchw/
Dall.cpp79 const int input_dim_h = src->info()->dimension(1); in convolve_nchw() local
122 const int in_h_end = std::min(in_h_end_t, input_dim_h); in convolve_nchw()
/external/ComputeLibrary/src/cpu/kernels/conv3d/neon/
Dlist.h57 const int input_dim_h = src->info()->dimension(2); in directconv3d_float_neon_ndhwc() local
110 const int in_h_end = std::min(in_h_end_t, input_dim_h); in directconv3d_float_neon_ndhwc()
Dquantized.h73 const int input_dim_h = src->info()->dimension(2); in directconv3d_quantized_neon_ndhwc() local
126 const int in_h_end = std::min(in_h_end_t, input_dim_h); in directconv3d_quantized_neon_ndhwc()