Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistributionKernels.cpp62 at::Tensor tmp_int_tensor; in bernoulli_scalar_kernel() local
64 tmp_int_tensor = self; in bernoulli_scalar_kernel()
66 tmp_int_tensor = at::empty(self.sizes(), self.options().dtype(at::kInt)); in bernoulli_scalar_kernel()
70 int *sample_int_ptr = tmp_int_tensor.data_ptr<int>(); in bernoulli_scalar_kernel()
96 OptionalTensorRef(self)->copy_(tmp_int_tensor); in bernoulli_scalar_kernel()