Searched refs:gInp_slice_C_ptr (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | GridSamplerKernel.cpp | 662 auto gInp_slice_C_ptr = (*gInp_slice_ptr)[c].data(); in backward() local
665 … mask_scatter_add(gInp_corner_arr, gInp_slice_C_ptr, i_gInp_nw_offset_arr, i_nw_mask_arr, len); in backward()
667 … mask_scatter_add(gInp_corner_arr, gInp_slice_C_ptr, i_gInp_ne_offset_arr, i_ne_mask_arr, len); in backward()
669 … mask_scatter_add(gInp_corner_arr, gInp_slice_C_ptr, i_gInp_sw_offset_arr, i_sw_mask_arr, len); in backward()
671 … mask_scatter_add(gInp_corner_arr, gInp_slice_C_ptr, i_gInp_se_offset_arr, i_se_mask_arr, len); in backward()
993 auto gInp_slice_C_ptr = (*gInp_slice_ptr)[c].data(); in backward() local
994 add_value_bounded(gInp_slice_C_ptr, len, xx, yy, gOut * coeff_x[i] * coeff_y[j]); in backward()
|