Searched defs:gGrid_ptr_NHW (Results 1 – 2 of 2) sorted by relevance
429 scalar_t *gGrid_ptr_NHW = grad_grid.data + index * gGrid_sW; in grid_sampler_2d_backward_kernel() local450 scalar_t *gGrid_ptr_NHW = grad_grid.data + index * gGrid_sW; in grid_sampler_2d_backward_kernel() local509 scalar_t *gGrid_ptr_NHW = grad_grid.data + index * gGrid_sW; in grid_sampler_2d_backward_kernel() local
771 scalar_t *gGrid_ptr_NHW = gGrid_ptr + n * gGrid_sN; in _grid_sampler_2d_cpu_fallback_backward() local