Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/utils/
H A Dbackend_registration.py119 def wrap_tensor_backend(self: torch.Tensor) -> bool: function
122 …return handle_torch_function(wrap_tensor_backend.__get__, (self,), self) # type: ignore[attr-defi…
126 wrap_tensor_backend.fget.__name__ = f'is_{custom_backend_name}' # type: ignore[attr-defined]
127 setattr(torch.Tensor, f'is_{custom_backend_name}', wrap_tensor_backend)
196 def wrap_tensor_backend(self: torch.nn.utils.rnn.PackedSequence) -> bool: function
200 setattr(torch.nn.utils.rnn.PackedSequence, f'is_{custom_backend_name}', wrap_tensor_backend)