Home
last modified time | relevance | path

Searched refs:_is_nested_tensor (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/distributed/checkpoint/
Doptimizer.py84 def _is_nested_tensor(val: torch.Tensor) -> bool: function
136 if _is_nested_tensor(value):