Searched refs:num_elems_read_per_iteration (Results 1 – 5 of 5) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/directconv2d/nhwc/neon/ |
| D | impl.cpp | 97 constexpr int num_elems_read_per_iteration = 16 / sizeof(T); in convolve_nhwc() local
162 …<= index_wc_end - num_elems_read_per_iteration; index_wc += num_elems_read_per_iteration, in_ptr_m… in convolve_nhwc()
225 …num_elems_read_per_iteration; index_c += num_elems_read_per_iteration, in_ptr_mover += num_elems_r… in convolve_nhwc()
|
| /external/ComputeLibrary/src/cpu/kernels/conv3d/neon/ |
| D | list.h | 48 constexpr int num_elems_read_per_iteration = 16 / sizeof(T); in directconv3d_float_neon_ndhwc() local
148 for(; index_c_in <= index_c_in_end - num_elems_read_per_iteration; in directconv3d_float_neon_ndhwc()
149 … index_c_in += num_elems_read_per_iteration, in_ptr_mover += num_elems_read_per_iteration) in directconv3d_float_neon_ndhwc()
153 … for(int k = 0; k < num_elems_read_per_iteration; ++k, weights_ptr_mover += index_c_out_end) in directconv3d_float_neon_ndhwc()
|
| D | quantized.h | 49 constexpr int num_elems_read_per_iteration = 16 / sizeof(T); in directconv3d_quantized_neon_ndhwc() local
169 for(; index_c_in <= index_c_in_end - num_elems_read_per_iteration; in directconv3d_quantized_neon_ndhwc()
170 … index_c_in += num_elems_read_per_iteration, in_ptr_mover += num_elems_read_per_iteration) in directconv3d_quantized_neon_ndhwc()
174 … for(int k = 0; k < num_elems_read_per_iteration; ++k, weights_ptr_mover += index_c_out_end) in directconv3d_quantized_neon_ndhwc()
|
| /external/ComputeLibrary/src/cpu/kernels/directconv2d/nchw/ |
| D | all.cpp | 108 constexpr int num_elems_read_per_iteration = 16 / sizeof(T); in convolve_nchw() local
148 …w <= ((in_w_end - num_elems_read_per_iteration)); index_w += num_elems_read_per_iteration, index_w… in convolve_nchw()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClCol2ImKernel.cpp | 73 constexpr unsigned int num_elems_read_per_iteration = 8; in validate_and_configure_window() local
76 Window win = calculate_max_window(*src, Steps(num_elems_read_per_iteration)); in validate_and_configure_window()
79 AccessWindowHorizontal input_access(src, 0, num_elems_read_per_iteration); in validate_and_configure_window()
|