Searched defs:out_H (Results 1 – 5 of 5) sorted by relevance
57 int64_t out_H = grid.size(2); in grid_sampler_3d_cpu_impl() local240 int64_t out_H = grid.size(2); in grid_sampler_3d_backward_cpu_impl() local470 int64_t out_H = grid.size(1); in _grid_sampler_2d_cpu_quantized() local574 int64_t out_H = grid.size(1); in _grid_sampler_2d_cpu_fallback() local741 int64_t out_H = grid.size(1); in _grid_sampler_2d_cpu_fallback_backward() local
160 size_t out_H = out_sizes[in_dim - 2]; in kernel_reduction_then_map_2d() local
39 index_t out_H = grid.sizes[1]; in grid_sampler_2d_kernel() local171 index_t out_H = grid.sizes[2]; in grid_sampler_3d_kernel() local328 index_t out_H = grid.sizes[1]; in grid_sampler_2d_backward_kernel() local536 index_t out_H = grid.sizes[2]; in grid_sampler_3d_backward_kernel() local
61 size_t out_H = out_sizes[2]; in conv2d_impl() local
1025 int64_t out_H = grid_slice.size(0); in grid_sample_2d_grid_slice_iterator() local