Searched refs:k_stride (Results 1 – 3 of 3) sorted by relevance
174 const uint32_t k_stride = round_up_po2(group_input_channels, kr); in create_deconvolution2d_nhwc() local177 …size_t packed_group_weights_size = (sizeof(float) * kernel_size * k_stride + sizeof(float)) * n_st… in create_deconvolution2d_nhwc()182 (sizeof(float) * kernel_size * k_stride + sizeof(float) * subkernels); in create_deconvolution2d_nhwc()201 subconvolution_params->w_stride = sizeof(float) + k_stride * subkernel_size * sizeof(float); in create_deconvolution2d_nhwc()235 const size_t zero_size = (k_stride << log2_input_element_size) + XNN_EXTRA_BYTES; in create_deconvolution2d_nhwc()
103 const size_t k_stride = round_up_po2(input_channels, kr); in create_fully_connected_nc() local105 …const size_t packed_weights_size = n_stride * (bias_element_size + (k_stride << log2_filter_elemen… in create_fully_connected_nc()
309 const size_t k_stride = round_up_po2(group_input_channels, kr); in create_convolution2d_nhwc() local311 …const size_t packed_group_weights_size = ((kernel_size * k_stride << log2_filter_element_size) + b… in create_convolution2d_nhwc()365 zero_size = XNN_EXTRA_BYTES + (k_stride << log2_input_element_size); in create_convolution2d_nhwc()