(
ctx,
means2d: Tensor,
ray_transforms: Tensor,
colors: Tensor,
opacities: Tensor,
normals: Tensor,
densify: Tensor,
backgrounds: Tensor,
masks: Tensor,
width: int,
height: int,
tile_size: int,
isect_offsets: Tensor,
flatten_ids: Tensor,
absgrad: bool,
distloss: bool,
)
| 1830 | |
| 1831 | @staticmethod |
| 1832 | def forward( |
| 1833 | ctx, |
| 1834 | means2d: Tensor, |
| 1835 | ray_transforms: Tensor, |
| 1836 | colors: Tensor, |
| 1837 | opacities: Tensor, |
| 1838 | normals: Tensor, |
| 1839 | densify: Tensor, |
| 1840 | backgrounds: Tensor, |
| 1841 | masks: Tensor, |
| 1842 | width: int, |
| 1843 | height: int, |
| 1844 | tile_size: int, |
| 1845 | isect_offsets: Tensor, |
| 1846 | flatten_ids: Tensor, |
| 1847 | absgrad: bool, |
| 1848 | distloss: bool, |
| 1849 | ) -> Tuple[Tensor, Tensor]: |
| 1850 | ( |
| 1851 | render_colors, |
| 1852 | render_alphas, |
| 1853 | render_normals, |
| 1854 | render_distort, |
| 1855 | render_median, |
| 1856 | last_ids, |
| 1857 | median_ids, |
| 1858 | ) = _make_lazy_cuda_func("rasterize_to_pixels_fwd_2dgs")( |
| 1859 | means2d, |
| 1860 | ray_transforms, |
| 1861 | colors, |
| 1862 | opacities, |
| 1863 | normals, |
| 1864 | backgrounds, |
| 1865 | masks, |
| 1866 | width, |
| 1867 | height, |
| 1868 | tile_size, |
| 1869 | isect_offsets, |
| 1870 | flatten_ids, |
| 1871 | ) |
| 1872 | |
| 1873 | ctx.save_for_backward( |
| 1874 | means2d, |
| 1875 | ray_transforms, |
| 1876 | colors, |
| 1877 | opacities, |
| 1878 | normals, |
| 1879 | densify, |
| 1880 | backgrounds, |
| 1881 | masks, |
| 1882 | isect_offsets, |
| 1883 | flatten_ids, |
| 1884 | render_colors, |
| 1885 | render_alphas, |
| 1886 | last_ids, |
| 1887 | median_ids, |
| 1888 | ) |
| 1889 | ctx.width = width |
nothing calls this directly
no test coverage detected