SYMBOL INDEX (177 symbols across 23 files)

FILE: build_extensions.py
  function get_commit_hash (line 15) | def get_commit_hash():
  function main (line 18) | def main():

FILE: gsplat/_helper.py
  function load_test_data (line 9) | def load_test_data(

FILE: gsplat/compression/png_compression.py
  class PngCompression (line 16) | class PngCompression:
    method _get_compress_fn (line 47) | def _get_compress_fn(self, param_name: str) -> Callable:
    method _get_decompress_fn (line 61) | def _get_decompress_fn(self, param_name: str) -> Callable:
    method compress (line 75) | def compress(self, compress_dir: str, splats: Dict[str, Tensor]) -> None:
    method decompress (line 113) | def decompress(self, compress_dir: str) -> Dict[str, Tensor]:
  function _crop_n_splats (line 135) | def _crop_n_splats(splats: Dict[str, Tensor], n_crop: int) -> Dict[str, ...
  function _compress_png (line 143) | def _compress_png(
  function _decompress_png (line 185) | def _decompress_png(compress_dir: str, param_name: str, meta: Dict[str, ...
  function _compress_png_16bit (line 215) | def _compress_png_16bit(
  function _decompress_png_16bit (line 263) | def _decompress_png_16bit(
  function _compress_npz (line 298) | def _compress_npz(
  function _decompress_npz (line 313) | def _decompress_npz(compress_dir: str, param_name: str, meta: Dict[str, ...
  function _compress_kmeans (line 322) | def _compress_kmeans(
  function _decompress_kmeans (line 391) | def _decompress_kmeans(

FILE: gsplat/compression/sort.py
  function sort_splats (line 7) | def sort_splats(splats: Dict[str, Tensor], verbose: bool = True) -> Dict...

FILE: gsplat/cuda/_backend.py
  function load_extension (line 23) | def load_extension(
  function cuda_toolkit_available (line 56) | def cuda_toolkit_available():
  function cuda_toolkit_version (line 65) | def cuda_toolkit_version():

FILE: gsplat/cuda/_torch_impl.py
  function _quat_scale_to_covar_preci (line 10) | def _quat_scale_to_covar_preci(
  function _persp_proj (line 54) | def _persp_proj(
  function _depth_compensation_from_cov3d (line 104) | def _depth_compensation_from_cov3d(cov3d: Tensor, eps2d: float) -> Tensor:
  function _lidar_proj (line 120) | def _lidar_proj(
  function _world_to_cam (line 168) | def _world_to_cam(
  function _compute_pix_velocity (line 193) | def _compute_pix_velocity(
  function _compute_lidar_velocity (line 227) | def _compute_lidar_velocity(p_view: Tensor, lin_vel: Tensor, ang_vel: Te...
  function _fully_fused_projection (line 265) | def _fully_fused_projection(
  function _fully_fused_lidar_projection (line 341) | def _fully_fused_lidar_projection(
  function _isect_tiles (line 422) | def _isect_tiles(
  function _isect_lidar_tiles (line 492) | def _isect_lidar_tiles(
  function _isect_offset_encode (line 585) | def _isect_offset_encode(isect_ids: Tensor, C: int, tile_width: int, til...
  function accumulate (line 610) | def accumulate(
  function accumulate_lidar (line 702) | def accumulate_lidar(
  function _rasterize_to_pixels (line 823) | def _rasterize_to_pixels(
  function _rasterize_to_points (line 921) | def _rasterize_to_points(
  function _eval_sh_bases_fast (line 1026) | def _eval_sh_bases_fast(basis_dim: int, dirs: Tensor):
  function _spherical_harmonics (line 1108) | def _spherical_harmonics(

FILE: gsplat/cuda/_wrapper.py
  function _make_lazy_cuda_func (line 7) | def _make_lazy_cuda_func(name: str) -> Callable:
  function spherical_harmonics (line 17) | def spherical_harmonics(
  function quat_scale_to_covar_preci (line 46) | def quat_scale_to_covar_preci(
  function persp_proj (line 80) | def persp_proj(
  function compute_pix_velocity (line 111) | def compute_pix_velocity(p_view: Tensor, # [C, N, 3]
  function world_to_cam (line 147) | def world_to_cam(
  function lidar_proj (line 175) | def lidar_proj(means: Tensor, # [C, N, 3]
  function compute_lidar_velocity (line 201) | def compute_lidar_velocity(p_view: Tensor, # [C, N, 3]
  function fully_fused_projection (line 230) | def fully_fused_projection(
  function fully_fused_lidar_projection (line 390) | def fully_fused_lidar_projection(
  function isect_tiles (line 533) | def isect_tiles(
  function isect_lidar_tiles (line 606) | def isect_lidar_tiles(
  function isect_offset_encode (line 680) | def isect_offset_encode(
  function map_points_to_lidar_tiles (line 699) | def map_points_to_lidar_tiles(
  function points_mapping_offset_encode (line 758) | def points_mapping_offset_encode(
  function populate_image_from_points (line 777) | def populate_image_from_points(
  function rasterize_to_pixels (line 811) | def rasterize_to_pixels(
  function rasterize_to_points (line 958) | def rasterize_to_points(
  function rasterize_to_indices_in_range (line 1122) | def rasterize_to_indices_in_range(
  function rasterize_to_indices_in_range_lidar (line 1208) | def rasterize_to_indices_in_range_lidar(
  class _QuatScaleToCovarPreci (line 1292) | class _QuatScaleToCovarPreci(torch.autograd.Function):
    method forward (line 1296) | def forward(
    method backward (line 1314) | def backward(ctx, v_covars: Tensor, v_precis: Tensor):
  class _PerspProj (line 1332) | class _PerspProj(torch.autograd.Function):
    method forward (line 1336) | def forward(
    method backward (line 1353) | def backward(ctx, v_means2d: Tensor, v_covars2d: Tensor):
  class _ComputePixVelocity (line 1368) | class _ComputePixVelocity(torch.autograd.Function):
    method forward (line 1372) | def forward(
    method backward (line 1391) | def backward(ctx, v_vel: Tensor):
  class _WorldToCam (line 1407) | class _WorldToCam(torch.autograd.Function):
    method forward (line 1411) | def forward(
    method backward (line 1424) | def backward(ctx, v_means_c: Tensor, v_covars_c: Tensor):
  class _LidarProj (line 1444) | class _LidarProj(torch.autograd.Function):
    method forward (line 1448) | def forward(
    method backward (line 1462) | def backward(ctx, v_means2d: Tensor, v_covars2d: Tensor, v_depth_compe...
  class _ComputeLidarVelocity (line 1475) | class _ComputeLidarVelocity(torch.autograd.Function):
    method forward (line 1479) | def forward(
    method backward (line 1493) | def backward(ctx, v_vel: Tensor):
  class _FullyFusedProjection (line 1504) | class _FullyFusedProjection(torch.autograd.Function):
    method forward (line 1508) | def forward(
    method backward (line 1574) | def backward(ctx, v_radii, v_means2d, v_depths, v_conics, v_compensati...
  class _FullyFusedLidarProjection (line 1651) | class _FullyFusedLidarProjection(torch.autograd.Function):
    method forward (line 1655) | def forward(
    method backward (line 1724) | def backward(ctx, v_radii, v_means2d, v_depths, v_conics, v_compensati...
  class _RasterizeToPixels (line 1805) | class _RasterizeToPixels(torch.autograd.Function):
    method forward (line 1809) | def forward(
    method backward (line 1868) | def backward(
  class _RasterizeToPoints (line 1948) | class _RasterizeToPoints(torch.autograd.Function):
    method forward (line 1952) | def forward(
    method backward (line 2028) | def backward(
  class _FullyFusedProjectionPacked (line 2125) | class _FullyFusedProjectionPacked(torch.autograd.Function):
    method forward (line 2129) | def forward(
    method backward (line 2206) | def backward(
  class _SphericalHarmonics (line 2339) | class _SphericalHarmonics(torch.autograd.Function):
    method forward (line 2343) | def forward(
    method backward (line 2353) | def backward(ctx, v_colors: Tensor):

FILE: gsplat/cuda/csrc/ext.cpp
  function PYBIND11_MODULE (line 4) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: gsplat/distributed.py
  function all_gather_int32 (line 10) | def all_gather_int32(
  function all_to_all_int32 (line 55) | def all_to_all_int32(
  function all_gather_tensor_list (line 102) | def all_gather_tensor_list(world_size: int, tensor_list: List[Tensor]) -...
  function all_to_all_tensor_list (line 170) | def all_to_all_tensor_list(
  function _find_free_port (line 260) | def _find_free_port():
  function _distributed_worker (line 272) | def _distributed_worker(
  function cli (line 304) | def cli(fn: Callable, args: Any, verbose: bool = False) -> bool:

FILE: gsplat/profile.py
  class timeit (line 11) | class timeit(object):
    method __init__ (line 32) | def __init__(self, name: str = "unnamed"):
    method __enter__ (line 37) | def __enter__(self):
    method __exit__ (line 42) | def __exit__(self, exc_type, exc_val, exc_tb):
    method __call__ (line 52) | def __call__(self, f: Callable) -> Callable:

FILE: gsplat/relocation.py
  function compute_relocation (line 10) | def compute_relocation(

FILE: gsplat/rendering.py
  function rasterization (line 22) | def rasterization(
  function lidar_rasterization (line 453) | def lidar_rasterization(
  function rasterization_inria_wrapper (line 832) | def rasterization_inria_wrapper(

FILE: gsplat/strategy/base.py
  class Strategy (line 8) | class Strategy:
    method check_sanity (line 15) | def check_sanity(
    method step_pre_backward (line 33) | def step_pre_backward(
    method step_post_backward (line 41) | def step_post_backward(

FILE: gsplat/strategy/default.py
  class DefaultStrategy (line 12) | class DefaultStrategy(Strategy):
    method initialize_state (line 96) | def initialize_state(self, scene_scale: float = 1.0) -> Dict[str, Any]:
    method check_sanity (line 112) | def check_sanity(
    method step_pre_backward (line 138) | def step_pre_backward(
    method step_post_backward (line 152) | def step_post_backward(
    method _update_state (line 203) | def _update_state(
    method _grow_gs (line 262) | def _grow_gs(
    method _prune_gs (line 312) | def _prune_gs(

FILE: gsplat/strategy/mcmc.py
  class MCMCStrategy (line 13) | class MCMCStrategy(Strategy):
    method initialize_state (line 57) | def initialize_state(self) -> Dict[str, Any]:
    method check_sanity (line 66) | def check_sanity(
    method step_post_backward (line 103) | def step_post_backward(
    method _relocate_gs (line 151) | def _relocate_gs(
    method _add_new_gs (line 174) | def _add_new_gs(

FILE: gsplat/strategy/ops.py
  function _multinomial_sample (line 14) | def _multinomial_sample(weights: Tensor, n: int, replacement: bool = Tru...
  function _update_param_with_optimizer (line 48) | def _update_param_with_optimizer(
  function duplicate (line 98) | def duplicate(
  function split (line 129) | def split(
  function remove (line 189) | def remove(
  function reset_opa (line 219) | def reset_opa(
  function relocate (line 250) | def relocate(
  function sample_add (line 310) | def sample_add(
  function inject_noise_to_position (line 357) | def inject_noise_to_position(

FILE: gsplat/utils.py
  function normalized_quat_to_rotmat (line 8) | def normalized_quat_to_rotmat(quat: Tensor) -> Tensor:
  function log_transform (line 36) | def log_transform(x):
  function inverse_log_transform (line 40) | def inverse_log_transform(y):
  function depth_to_points (line 44) | def depth_to_points(
  function depth_to_normal (line 108) | def depth_to_normal(
  function get_projection_matrix (line 134) | def get_projection_matrix(znear, zfar, fovX, fovY, device="cuda"):

FILE: setup.py
  function get_ext (line 19) | def get_ext():
  function get_extensions (line 25) | def get_extensions():

FILE: tests/_test_distributed.py
  function _main_all_gather_int32 (line 13) | def _main_all_gather_int32(local_rank: int, world_rank: int, world_size:...
  function test_all_gather_int32 (line 28) | def test_all_gather_int32():
  function _main_all_to_all_int32 (line 32) | def _main_all_to_all_int32(local_rank: int, world_rank: int, world_size:...
  function test_all_to_all_int32 (line 47) | def test_all_to_all_int32():
  function _main_all_gather_tensor_list (line 51) | def _main_all_gather_tensor_list(local_rank: int, world_rank: int, world...
  function test_all_gather_tensor_list (line 71) | def test_all_gather_tensor_list():
  function _main_all_to_all_tensor_list (line 75) | def _main_all_to_all_tensor_list(local_rank: int, world_rank: int, world...
  function test_all_to_all_tensor_list (line 106) | def test_all_to_all_tensor_list():

FILE: tests/test_basic.py
  function test_data (line 20) | def test_data():
  function test_quat_scale_to_covar_preci (line 60) | def test_quat_scale_to_covar_preci(test_data, triu: bool):
  function test_world_to_cam (line 98) | def test_world_to_cam(test_data):
  function test_persp_proj (line 136) | def test_persp_proj(test_data):
  function test_lidar_proj (line 174) | def test_lidar_proj(test_data, calc_depth_comp_grads: bool):
  function test_compute_lidar_velocity (line 211) | def test_compute_lidar_velocity(test_data):
  function test_compute_pix_velocity (line 247) | def test_compute_pix_velocity(test_data):
  function test_projection (line 288) | def test_projection(test_data, fused: bool, calc_compensations: bool, us...
  function test_fully_fused_projection_packed (line 403) | def test_fully_fused_projection_packed(test_data, fused: bool, sparse_gr...
  function test_lidar_projection (line 568) | def test_lidar_projection(test_data, fused: bool, calc_compensations: bo...
  function test_isect (line 698) | def test_isect(test_data):
  function test_isect_lidar (line 727) | def test_isect_lidar(test_data):
  function test_map_points_to_lidar_tiles (line 787) | def test_map_points_to_lidar_tiles(test_data):
  function test_populate_image_from_points (line 844) | def test_populate_image_from_points(test_data):
  function test_rasterize_to_pixels (line 893) | def test_rasterize_to_pixels(test_data, channels: int):
  function test_rasterize_to_points (line 1016) | def test_rasterize_to_points(test_data, compute_alpha_sum_until_points: ...
  function test_accumulate_until_points (line 1203) | def test_accumulate_until_points(test_data, use_random_test_points: bool...
  function test_sh (line 1376) | def test_sh(test_data, sh_degree: int):

FILE: tests/test_compression.py
  function test_png_compression (line 16) | def test_png_compression():

FILE: tests/test_rasterization.py
  function test_rasterization (line 23) | def test_rasterization(per_view_color: bool, sh_degree: Optional[int], r...
  function test_lidar_rasterization (line 86) | def test_lidar_rasterization(channels: int):

FILE: tests/test_strategy.py
  function test_strategy (line 16) | def test_strategy():