Searched defs:out_W (Results 1 – 5 of 5) sorted by relevance
58 int64_t out_W = grid.size(3); in grid_sampler_3d_cpu_impl() local241 int64_t out_W = grid.size(3); in grid_sampler_3d_backward_cpu_impl() local471 int64_t out_W = grid.size(2); in _grid_sampler_2d_cpu_quantized() local575 int64_t out_W = grid.size(2); in _grid_sampler_2d_cpu_fallback() local742 int64_t out_W = grid.size(2); in _grid_sampler_2d_cpu_fallback_backward() local
163 size_t out_W = out_sizes[in_dim - 1]; in kernel_reduction_then_map_2d() local
40 index_t out_W = grid.sizes[2]; in grid_sampler_2d_kernel() local172 index_t out_W = grid.sizes[3]; in grid_sampler_3d_kernel() local329 index_t out_W = grid.sizes[2]; in grid_sampler_2d_backward_kernel() local537 index_t out_W = grid.sizes[3]; in grid_sampler_3d_backward_kernel() local
65 size_t out_W = out_sizes[3]; in conv2d_impl() local
1026 int64_t out_W = grid_slice.size(1); in grid_sample_2d_grid_slice_iterator() local