Searched refs:normalize_to_torch_size (Results 1 – 3 of 3) sorted by relevance
26 normalize_to_torch_size,1002 torch_size = normalize_to_torch_size(size)1045 torch_size = normalize_to_torch_size(size)1091 torch_size = normalize_to_torch_size(size)1136 torch_size = normalize_to_torch_size(size)1180 torch_size = normalize_to_torch_size(size)1221 torch_size = normalize_to_torch_size(size)
300 def normalize_to_torch_size(size) -> torch.Size: # type: ignore[no-untyped-def] function
29 from torch.distributed.tensor._utils import normalize_to_torch_size802 normalized_size = normalize_to_torch_size(normalized_shape)901 normalized_size = normalize_to_torch_size(normalized_shape)