Searched refs:contiguous_out_arg (Results 1 – 1 of 1) sorted by relevance
21 static c10::MaybeOwned<Tensor> contiguous_out_arg(const Tensor &tensor) { in contiguous_out_arg() function34 auto values_ = contiguous_out_arg(values); in cummax_helper_cuda()35 auto indices_ = contiguous_out_arg(indices); in cummax_helper_cuda()51 auto values_ = contiguous_out_arg(values); in cummin_helper_cuda()52 auto indices_ = contiguous_out_arg(indices); in cummin_helper_cuda()78 auto result_ = contiguous_out_arg(result); in _logcumsumexp_out_cuda()97 auto result_ = contiguous_out_arg(result); in cumsum_cuda_kernel()105 auto result_ = contiguous_out_arg(result); in cumprod_cuda_kernel()