Searched defs:alignment_size (Results 1 – 2 of 2) sorted by relevance
119 def get_padded_length(x: Union[int, torch.SymInt], alignment_size) -> int: argument
1734 def pad_last_dim(input_tensor, alignment_size, slice: bool = False): argument