Searched refs:tensor_is_aligned (Results 1 – 2 of 2) sorted by relevance
48 tensor_is_aligned,929 if i in static_input_idxs and tensor_is_aligned(input):
1884 def tensor_is_aligned(tensor: torch.Tensor): function1904 return config.assume_aligned_inputs or tensor_is_aligned(example_input)