Searched defs:grid_offset (Results 1 – 5 of 5) sorted by relevance
58 const index_t grid_offset = n * grid_sN + h * grid_sH + w * grid_sW; in grid_sampler_2d_kernel() local194 const index_t grid_offset = n * grid_sN + d * grid_sD + h * grid_sH + w * grid_sW; in grid_sampler_3d_kernel() local359 const auto grid_offset = n * grid_sN + h * grid_sH + w * grid_sW; in grid_sampler_2d_backward_kernel() local573 const auto grid_offset = n * grid_sN + d * grid_sD + h * grid_sH + w * grid_sW; in grid_sampler_3d_backward_kernel() local
64 const std::vector<size_t> &grid_offset, in launch()194 const std::vector<size_t> &grid_offset) { in bind()
101 grid_offset, enumerator
324 auto grid_offset = validate_grid_offset(q, dims, d_grid_offset); in clEnqueueNDRangeKernel() local
1049 auto grid_offset = spatial_offset * 2; in grid_sample_2d_grid_slice_iterator() local