Searched defs:inp_H (Results 1 – 3 of 3) sorted by relevance
54 int64_t inp_H = input.size(3); in grid_sampler_3d_cpu_impl() local237 int64_t inp_H = input.size(3); in grid_sampler_3d_backward_cpu_impl() local468 int64_t inp_H = input.size(2); in _grid_sampler_2d_cpu_quantized() local572 int64_t inp_H = input.size(2); in _grid_sampler_2d_cpu_fallback() local739 int64_t inp_H = input.size(2); in _grid_sampler_2d_cpu_fallback_backward() local
37 index_t inp_H = input.sizes[2]; in grid_sampler_2d_kernel() local168 index_t inp_H = input.sizes[3]; in grid_sampler_3d_kernel() local326 index_t inp_H = input.sizes[2]; in grid_sampler_2d_backward_kernel() local533 index_t inp_H = input.sizes[3]; in grid_sampler_3d_backward_kernel() local
465 const int64_t inp_H; member708 const int64_t inp_H; member816 const int64_t inp_H; member