Searched refs:inputPointY (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | ReplicationPadding.cu | 111 …const int inputPointY = imin(imax(padT, outputPointY), input.size(2) + padT - 1) - oStartY + iStar… in replication_pad_forward_kernel2d() local
113 scalar_t valueToCopy = input[batch][plane][inputPointY][inputPointX]; in replication_pad_forward_kernel2d()
140 …const int inputPointY = imin(imax(padT, outputPointY), gradInput.size(2) + padT - 1) - oStartY + i… in replication_pad_backward_kernel() local
143 gpuAtomicAddNoReturn(&gradInput[batch][plane][inputPointY][inputPointX], valueToCopy); in replication_pad_backward_kernel()
175 const int inputPointY = imin(imax(ptop, outputPointY), in replication_pad_forward_kernel3d() local
181 input[batch][plane][inputPointZ][inputPointY][inputPointX]; in replication_pad_forward_kernel3d()
217 const int inputPointY = imin(imax(ptop, outputPointY), in replication_pad_backward_kernel() local
224 gpuAtomicAddNoReturn(&gradInput[batch][plane][inputPointZ][inputPointY][inputPointX], in replication_pad_backward_kernel()
|