Searched defs:grad_theta (Results 1 – 2 of 2) sorted by relevance
111 auto grad_theta = base_grid.view({N, H * W, 3}) in affine_grid_generator_4D_backward() local127 auto grad_theta = base_grid.view({N, D * H * W, 4}) in affine_grid_generator_5D_backward() local
33 const Tensor& grad_theta, in cudnn_affine_grid_generator_backward()