Searched refs:resize_impl_cuda_ (Results 1 – 4 of 4) sorted by relevance
37 at::native::resize_impl_cuda_(result.unsafeGetTensorImpl(), size, stride_opt); in set_storage_cuda_()
29 inline TensorImpl* resize_impl_cuda_( in resize_impl_cuda_() function
58 resize_impl_cuda_(self_, size, /*strides=*/std::nullopt); in resize_cuda_()
124 resize_impl_cuda_(out.unsafeGetTensorImpl(), self.sizes(), self.strides()); in batch_norm_elementwise()