Searched defs:inputPointX (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | ReplicationPadding.cu | 58 …const auto inputPointX = imin(imax(padL, outputPointX), input.size(2) + padL - 1) - oStartX + iSta… in replication_pad_forward_kernel1d() local 82 …const auto inputPointX = imin(imax(padL, outputPointX), gradInput.size(2) + padL - 1) - oStartX + … in replication_pad_backward_kernel() local 110 …const int inputPointX = imin(imax(padL, outputPointX), input.size(3) + padL - 1) - oStartX + iStar… in replication_pad_forward_kernel2d() local 139 …const int inputPointX = imin(imax(padL, outputPointX), gradInput.size(3) + padL - 1) - oStartX + i… in replication_pad_backward_kernel() local 173 const int inputPointX = imin(imax(pleft, outputPointX), in replication_pad_forward_kernel3d() local 215 const int inputPointX = imin(imax(pleft, outputPointX), in replication_pad_backward_kernel() local
|