Searched refs:conv_dst_ch_aligned (Results 1 – 1 of 1) sorted by relevance
34 int conv_dst_ch_aligned = AlignByN(conv_attr.weights.shape.o, 4); in UploadWeights() local35 int conv_weights_count = conv_src_ch_aligned * conv_dst_ch_aligned; in UploadWeights()37 gpu_data.reserve(dw_dst_ch_aligned + dw_weights_count + conv_dst_ch_aligned + in UploadWeights()65 for (int i = 0; i < conv_dst_ch_aligned; ++i) { in UploadWeights()73 for (int d = 0; d < conv_dst_ch_aligned / 4; ++d) { in UploadWeights()