Searched defs:base_grid (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AffineGridGenerator.cpp | 37 auto base_grid = at::empty({N, H, W, 3}, theta.options()); in make_base_grid_4D() local
54 auto base_grid = at::empty({N, D, H, W, 4}, theta.options()); in make_base_grid_5D() local
71 Tensor base_grid = make_base_grid_4D(theta, N, C, H, W, align_corners); in affine_grid_generator_4D() local
84 Tensor base_grid = make_base_grid_5D(theta, N, C, D, H, W, align_corners); in affine_grid_generator_5D() local
109 auto base_grid = make_base_grid_4D(grad_grid, N, C, H, W, align_corners); in affine_grid_generator_4D_backward() local
125 auto base_grid = make_base_grid_5D(grad_grid, N, C, D, H, W, align_corners); in affine_grid_generator_5D_backward() local
|