Searched refs:devInput (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | ReplicationPadding.cu | 449 auto devInput = input_.packed_accessor64<const scalar_t, 3>(); in TORCH_IMPL_FUNC() local
465 at::cuda::getCurrentCUDAStream()>>>(devInput, devOutput, padL, block_y, block_z); in TORCH_IMPL_FUNC()
554 auto devInput = input_.packed_accessor64<const scalar_t, 4>(); in TORCH_IMPL_FUNC() local
566 devInput, devOutput, padT, padL, block_y, block_z); in TORCH_IMPL_FUNC()
647 auto devInput = input_.packed_accessor64<const scalar_t, 5>(); in TORCH_IMPL_FUNC() local
663 devInput, devOutput, pfront, ptop, pleft, block_y, block_z); in TORCH_IMPL_FUNC()
|
| D | FractionalMaxPool2d.cu | 189 auto devInput = input_.packed_accessor64<const scalar_t, 4>(); in TORCH_IMPL_FUNC() local
195 devOutput, devIndices, devInput, devSamples, in TORCH_IMPL_FUNC()
|