Searched refs:_create_nested_tensor_from_mask (Results 1 – 1 of 1) sorted by relevance
2769 def _create_nested_tensor_from_mask(self, tensor_device, requires_grad=False): member in TestNestedTensorAutograd2818 nt = self._create_nested_tensor_from_mask(device)2823 nt_1 = self._create_nested_tensor_from_mask(device)2824 nt_2 = self._create_nested_tensor_from_mask(device)2831 grad_output = self._create_nested_tensor_from_mask(device)2839 nt_1 = self._create_nested_tensor_from_mask(device)2840 nt_2 = self._create_nested_tensor_from_mask(device)2849 grad_output = self._create_nested_tensor_from_mask(device)