Searched refs:devOutput (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | ReplicationPadding.cu | 450 auto devOutput = output_.packed_accessor64<scalar_t, 3>(); in TORCH_IMPL_FUNC() local
452 int64_t outputPlaneSize = devOutput.size(2); in TORCH_IMPL_FUNC()
453 int64_t size1 = devOutput.size(1); in TORCH_IMPL_FUNC()
454 int64_t size0 = devOutput.size(0); in TORCH_IMPL_FUNC()
465 at::cuda::getCurrentCUDAStream()>>>(devInput, devOutput, padL, block_y, block_z); in TORCH_IMPL_FUNC()
555 auto devOutput = output_.packed_accessor64<scalar_t, 4>(); in TORCH_IMPL_FUNC() local
556 int64_t outputPlaneSize = devOutput.size(2) * devOutput.size(3); in TORCH_IMPL_FUNC()
557 int64_t size1 = devOutput.size(1); in TORCH_IMPL_FUNC()
558 int64_t size0 = devOutput.size(0); in TORCH_IMPL_FUNC()
566 devInput, devOutput, padT, padL, block_y, block_z); in TORCH_IMPL_FUNC()
[all …]
|
| D | FractionalMaxPool2d.cu | 190 auto devOutput = output_.packed_accessor64<scalar_t, 4>(); in TORCH_IMPL_FUNC() local
195 devOutput, devIndices, devInput, devSamples, in TORCH_IMPL_FUNC()
|