Searched refs:devGradInput (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReplicationPadding.cu | 280 auto devGradInput = gradInput_.packed_accessor64<scalar_t, 4>(); in replication_pad2d_backward_out_cuda_template() local
296 devGradInput, devGradOutput, padT, padL, block_y, block_z); in replication_pad2d_backward_out_cuda_template()
394 auto devGradInput = gradInput_.packed_accessor64<scalar_t, 5>(); in replication_pad3d_backward_out_cuda_template() local
410 devGradInput, devGradOutput, pfront, ptop, pleft, block_y, block_z); in replication_pad3d_backward_out_cuda_template()
511 auto devGradInput = gradInput_.packed_accessor64<scalar_t, 3>(); in TORCH_IMPL_FUNC() local
527 devGradInput, devGradOutput, padL, block_y, block_z); in TORCH_IMPL_FUNC()
|
| H A D | FractionalMaxPool2d.cu | 264 auto devGradInput = gradInput_.packed_accessor64<scalar_t, 4>(); in TORCH_IMPL_FUNC() local
268 devGradInput, devGradOutput, devIndices); in TORCH_IMPL_FUNC()
|