Home
last modified time | relevance | path

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

/external/pytorch/torch/distributed/
D_functional_collectives.py240 res = _FromTorchTensor.apply(tensor)
328 res = _FromTorchTensor.apply(tensor)
526 return _FromTorchTensor.apply(tensor)
780 class _FromTorchTensor(torch.autograd.Function): class
/external/pytorch/torch/distributed/tensor/
D_api.py119 class _FromTorchTensor(torch.autograd.Function): class
422 return _FromTorchTensor.apply( # pyre-ignore[16]: autograd func