Searched refs:mb_unwrap_functional_tensor (Results 1 – 3 of 3) sorted by relevance
17 from torch._subclasses.functional_tensor import mb_unwrap_functional_tensor20 tensor = mb_unwrap_functional_tensor(tensor)
522 mb_unwrap_functional_tensor,527 tgt = mb_unwrap_functional_tensor(new_offsets)528 src = mb_unwrap_functional_tensor(inp._offsets)
804 def mb_unwrap_functional_tensor(tensor: torch.Tensor): function