Searched defs:_RefType (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/distributed/_tools/ |
| D | mem_tracker.py | 50 class _RefType(str, Enum): class
140 self, size: int, element_size: int, device: torch.device, reftype: _RefType
218 reftype: _RefType,
431 old_reftype: Optional[_RefType] = None,
473 reftype: _RefType,
553 def _track(self, reftype: _RefType, t: torch.Tensor) -> None:
770 self, reftype: _RefType, optimizer: optim.Optimizer
|