Searched refs:tmp_tensor (Results 1 – 4 of 4) sorted by relevance
131 at::Tensor tmp_tensor; in exponential_kernel() local134 tmp_tensor = self; in exponential_kernel()136 tmp_tensor = at::empty(self.sizes(), self.options().dtype(at::kDouble)); in exponential_kernel()138 tmp_tensor = at::empty(self.sizes(), self.options().dtype(at::kFloat)); in exponential_kernel()143 tmp_scalar_t *sample_ptr = tmp_tensor.data_ptr<tmp_scalar_t>(); in exponential_kernel()188 self.copy_(tmp_tensor); in exponential_kernel()
590 tmp_tensor = torch.tensor(595 tmp_tensor = torch.tensor(sequence, dtype=dtype, device=device)597 return tmp_tensor._typed_storage()._untyped_storage939 tmp_tensor = torch.tensor(942 tmp_tensor.set_(948 tmp_tensor = torch.tensor(952 tmp_tensor[idx] = value997 tmp_tensor = torch.tensor(1000 return tmp_tensor[idx_wrapped].item()
586 Tensor tmp_tensor = output.contiguous(); in _rrelu_with_noise_train() local587 scalar_t* output_data = tmp_tensor.data_ptr<scalar_t>(); in _rrelu_with_noise_train()604 output.copy_(tmp_tensor); in _rrelu_with_noise_train()
286 tmp_tensor = torch.empty(application, dtype=torch.int8, device="cuda")287 del tmp_tensor