Searched defs:new_grid (Results 1 – 1 of 1) sorted by relevance
122 auto new_grid = reshape_dim_into(*grid_bdim, 1, grid); in grid_sample_batch_rule() local128 auto new_grid = reshape_dim_into(*grid_bdim, 0, grid); in grid_sample_batch_rule() local190 auto new_grid = std::get<2>(new_bw_input); in grid_sample_backward_batch_rule() local210 auto new_grid = std::get<2>(new_bw_input); in cudnn_grid_sample_backward_batch_rule() local