Searched refs:weights_stride (Results 1 – 1 of 1) sorted by relevance
584 …const size_t weights_stride = (kernel_size * k_stride << log2_filter_element_size) + bias_element_… in create_convolution2d_nhwc() local588 gemm_parameters->nr * weights_stride, in create_convolution2d_nhwc()591 group_weights = (void*) ((uintptr_t) group_weights + n_stride * weights_stride); in create_convolution2d_nhwc()