Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DUpSampleNearest2d.cu403 …Tensor grad_input_c = grad_input.is_contiguous() ? grad_input : at::empty(grad_input.sizes(), grad… in upsample_nearest2d_backward_out_cuda_template() local
419 auto idata = grad_input_c.mutable_data_ptr<scalar_t>(); in upsample_nearest2d_backward_out_cuda_template()
439 grad_input.copy_(grad_input_c); in upsample_nearest2d_backward_out_cuda_template()
DUpSampleBilinear2d.cu436 …Tensor grad_input_c = grad_input.is_contiguous() ? grad_input : at::zeros(grad_input.sizes(), grad… in upsample_bilinear2d_backward_out_cuda_template() local
439 auto idata = grad_input_c.mutable_data_ptr<scalar_t>(); in upsample_bilinear2d_backward_out_cuda_template()
465 grad_input.copy_(grad_input_c); in upsample_bilinear2d_backward_out_cuda_template()