Searched refs:num_input_slices2pad (Results 1 – 1 of 1) sorted by relevance
89 auto num_input_slices2pad = (4 - num_input_channels % 4) % 4; in prepack() local90 if (num_output_slices2pad != 0 || num_input_slices2pad != 0) { in prepack()93 …weight = at::pad(weight, {0, 0, 0, 0, 0, num_input_slices2pad, 0, num_output_slices2pad}, "constan… in prepack()