Searched refs:grad_input_memory_span (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cu | 322 const index_t grad_input_memory_span, in grid_sampler_2d_backward_kernel() argument
396 ….data, iy_nw, ix_nw, gInp_sH, gInp_sW, inp_H, inp_W, nw * gOut, NC_offset, grad_input_memory_span); in grid_sampler_2d_backward_kernel()
397 ….data, iy_ne, ix_ne, gInp_sH, gInp_sW, inp_H, inp_W, ne * gOut, NC_offset, grad_input_memory_span); in grid_sampler_2d_backward_kernel()
398 ….data, iy_sw, ix_sw, gInp_sH, gInp_sW, inp_H, inp_W, sw * gOut, NC_offset, grad_input_memory_span); in grid_sampler_2d_backward_kernel()
399 ….data, iy_se, ix_se, gInp_sH, gInp_sW, inp_H, inp_W, se * gOut, NC_offset, grad_input_memory_span); in grid_sampler_2d_backward_kernel()
442 …st, ix_nearest, gInp_sH, gInp_sW, inp_H, inp_W, *gOut_ptr_NCHW, NC_offset, grad_input_memory_span); in grid_sampler_2d_backward_kernel()
496 grad_input_memory_span); in grid_sampler_2d_backward_kernel()
528 const index_t grad_input_memory_span, in grid_sampler_3d_backward_kernel() argument
646 NC_offset, grad_input_memory_span); in grid_sampler_3d_backward_kernel()
648 NC_offset, grad_input_memory_span); in grid_sampler_3d_backward_kernel()
[all …]
|