Searched refs:log2_element_size (Results 1 – 3 of 3) sorted by relevance
25 uint32_t log2_element_size);32 uint32_t log2_element_size);37 uint32_t log2_element_size);42 uint32_t log2_element_size);48 uint32_t log2_element_size);65 uint32_t log2_element_size);
21 uint32_t log2_element_size) in xnn_indirection_init_conv2d() argument26 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_conv2d()83 uint32_t log2_element_size) in xnn_indirection_init_dwconv2d() argument87 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_dwconv2d()136 uint32_t log2_element_size) in xnn_indirection_init_deconv2d() argument140 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_deconv2d()191 uint32_t log2_element_size) in xnn_indirection_init_subconv2d() argument196 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_subconv2d()257 uint32_t log2_element_size) in xnn_indirection_init_maxpool2d() argument261 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_maxpool2d()[all …]
139 uint32_t log2_element_size, in setup_channel_shuffle_nc() argument161 .x_stride = channel_shuffle_op->input_pixel_stride << log2_element_size, in setup_channel_shuffle_nc()163 .y_stride = channel_shuffle_op->output_pixel_stride << log2_element_size, in setup_channel_shuffle_nc()164 .n = channel_shuffle_op->group_channels << log2_element_size, in setup_channel_shuffle_nc()