Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReplicationPadding.cu281 auto devGradOutput = gradOutput_.packed_accessor64<const scalar_t, 4>(); in replication_pad2d_backward_out_cuda_template() local
283 int64_t outputPlaneSize = devGradOutput.size(2) * devGradOutput.size(3); in replication_pad2d_backward_out_cuda_template()
284 int64_t size1 = devGradOutput.size(1); in replication_pad2d_backward_out_cuda_template()
285 int64_t size0 = devGradOutput.size(0); in replication_pad2d_backward_out_cuda_template()
296 devGradInput, devGradOutput, padT, padL, block_y, block_z); in replication_pad2d_backward_out_cuda_template()
395 auto devGradOutput = gradOutput_.packed_accessor64<const scalar_t, 5>(); in replication_pad3d_backward_out_cuda_template() local
397 …const int64_t outputPlaneSize = devGradOutput.size(2) * devGradOutput.size(3) * devGradOutput.size… in replication_pad3d_backward_out_cuda_template()
398 const int64_t size1 = devGradOutput.size(1); in replication_pad3d_backward_out_cuda_template()
399 const int64_t size0 = devGradOutput.size(0); in replication_pad3d_backward_out_cuda_template()
410 devGradInput, devGradOutput, pfront, ptop, pleft, block_y, block_z); in replication_pad3d_backward_out_cuda_template()
[all …]
H A DFractionalMaxPool2d.cu265 auto devGradOutput = gradOutput_.packed_accessor64<const scalar_t, 4>(); in TORCH_IMPL_FUNC() local
268 devGradInput, devGradOutput, devIndices); in TORCH_IMPL_FUNC()