Searched defs:cast_fn (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/ | ||
| D | utils.py | 71 def cast_fn(x: torch.Tensor) -> torch.Tensor: function |
| /external/pytorch/torch/distributed/fsdp/ | ||
| D | _common_utils.py | 509 def cast_fn( function |