Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DScanKernels.h10 void launch_cummax_cuda_kernel(const TensorBase& self, const TensorBase& values,
DCumminmaxKernel.cu13 void launch_cummax_cuda_kernel(const TensorBase& self, const TensorBase& values, const TensorBase& … in launch_cummax_cuda_kernel() function
DScanKernels.cpp36 launch_cummax_cuda_kernel(self, *values_, *indices_, dim); in cummax_helper_cuda()