Searched refs:upcast_tensor (Results 1 – 1 of 1) sorted by relevance
346 def upcast_tensor(x, dtype=torch.float32): function876 upcast = partial(upcast_tensor, dtype=torch.float64)