Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReplicationPadding.cu65 __global__ void replication_pad_backward_kernel( in replication_pad_backward_kernel() function
118 __global__ void replication_pad_backward_kernel( in replication_pad_backward_kernel() function
186 __global__ void replication_pad_backward_kernel( in replication_pad_backward_kernel() function
295replication_pad_backward_kernel <<<gridSize, blockSize, 0, at::cuda::getCurrentCUDAStream()>>>( in replication_pad2d_backward_out_cuda_template()
409replication_pad_backward_kernel <<<gridSize, blockSize, 0, at::cuda::getCurrentCUDAStream()>>>( in replication_pad3d_backward_out_cuda_template()
526replication_pad_backward_kernel <<<gridSize, blockSize, 0, at::cuda::getCurrentCUDAStream()>>>( in TORCH_IMPL_FUNC()