Searched refs:last_dim_padding (Results 1 – 1 of 1) sorted by relevance
430 def pad_if_necessary(t, name, last_dim_padding): argument436 [[0, 0] for _ in range(len(t.shape) - 1)] + [last_dim_padding])