Searched defs:tensor_device (Results 1 – 3 of 3) sorted by relevance
42 def tensor_device(types, args=(), kwargs=None, pg=None): function
2753 def _create_leaf_nested_tensor_from_list(self, tensor_device, requires_grad=False): argument2760 def _create_nested_tensor_from_list(self, tensor_device, requires_grad=False): argument2769 def _create_nested_tensor_from_mask(self, tensor_device, requires_grad=False): argument
662 const BackendDevice& tensor_device = tensor->GetDevice(); in ExtractIRAndPrepareTensorData() local