Searched defs:gOut_sH (Results 1 – 2 of 2) sorted by relevance
340 index_t gOut_sH = grad_output.strides[2]; in grid_sampler_2d_backward_kernel() local551 index_t gOut_sH = grad_output.strides[3]; in grid_sampler_3d_backward_kernel() local
255 int64_t gOut_sH = grad_output.stride(3); in grid_sampler_3d_backward_cpu_impl() local753 int64_t gOut_sH = grad_output.stride(2); in _grid_sampler_2d_cpu_fallback_backward() local