Searched refs:kernel_dim_h (Results 1 – 4 of 4) sorted by relevance
77 const int kernel_dim_h = weights->info()->dimension(2); in convolve_nhwc() local129 const int in_h_end_t = in_h_start_t + kernel_dim_h; in convolve_nhwc()141 const int index_h_end = kernel_dim_h - (in_h_end_t - in_h_end); in convolve_nhwc()191 const int in_h_end_t = in_h_start_t + kernel_dim_h; in convolve_nhwc()203 const int wei_h_end = kernel_dim_h - (in_h_end_t - in_h_end); in convolve_nhwc()
88 const int kernel_dim_h = weights->info()->dimension(1); in convolve_nchw() local116 const int in_h_end_t = in_h_start_t + kernel_dim_h; in convolve_nchw()127 const int wei_h_end = kernel_dim_h - (in_h_end_t - in_h_end); in convolve_nchw()
65 const int kernel_dim_h = weights->info()->dimension(3); in directconv3d_float_neon_ndhwc() local102 const int in_h_end_t = in_h_start_t + kernel_dim_h; in directconv3d_float_neon_ndhwc()118 const int wei_h_end = kernel_dim_h - (in_h_end_t - in_h_end); in directconv3d_float_neon_ndhwc()
81 const int kernel_dim_h = weights->info()->dimension(3); in directconv3d_quantized_neon_ndhwc() local118 const int in_h_end_t = in_h_start_t + kernel_dim_h; in directconv3d_quantized_neon_ndhwc()134 const int wei_h_end = kernel_dim_h - (in_h_end_t - in_h_end); in directconv3d_quantized_neon_ndhwc()