MCPcopy Create free account

hub / github.com/VAST-AI-Research/TriplaneGaussian / functions

Functions315 in github.com/VAST-AI-Research/TriplaneGaussian

↓ 1 callersMethodset_use_memory_efficient_attention_xformers
( self, valid: bool, attention_op: Optional[Callable] = None )
tgs/models/tokenizers/dinov2.py:307
↓ 1 callersFunctionsquare_distance
Calculate Euclid distance between each two points. src^T * dst = xn * xm + yn * ym + zn * zm; sum(src^2, dim=-1) = xn*xn + yn*yn + zn*zn
tgs/models/snowflake/utils.py:266
↓ 1 callersFunctionthree_interpolate
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/interpolate.cpp:42
↓ 1 callersFunctionthree_nn
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/interpolate.cpp:14
↓ 1 callersMethodupdate_step
(self, epoch: int, global_step: int, on_load_weights: bool = False)
tgs/utils/base.py:49
↓ 1 callersMethodupdate_step_end
(self, epoch: int, global_step: int)
tgs/utils/base.py:55
FunctionPYBIND11_MODULE
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/bindings.cpp:6
Method__call__
(self, module, unused_inputs)
tgs/models/snowflake/utils.py:593
Method__getitem__
(self, index)
tgs/data.py:216
Method__init__
(self, cfg)
infer.py:55
Method__init__
(self, cfg: Any)
tgs/data.py:66
Method__init__
(self, cfg: Optional[dict] = None)
tgs/utils/base.py:15
Method__init__
( self, cfg: Optional[Union[dict, DictConfig]] = None, *args, **kwargs )
tgs/utils/base.py:96
Method__init__
(self, w2c, intrinsic, FoVx, FoVy, height, width, trans=np.array([0.0, 0.0, 0.0]), scale=1.0)
tgs/models/renderer.py:63
Method__init__
(self, query_dim: int, context_dim: int, n_heads: int, d_head: int)
tgs/models/transformers.py:103
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
tgs/models/transformers.py:169
Method__init__
( self, dim: int, dim_out: Optional[int] = None, mult: int = 4, dropou
tgs/models/transformers.py:413
Method__init__
(self, dim_in: int, dim_out: int)
tgs/models/transformers.py:491
Method__init__
(self, dim_in: int, dim_out: int)
tgs/models/transformers.py:519
Method__init__
(self, embedding_dim: int, num_embeddings: int)
tgs/models/transformers.py:537
Method__init__
(self, embedding_dim: int, condition_dim: int)
tgs/models/transformers.py:559
Method__init__
(self, embedding_dim: int, condition_dim: int, zero_init: bool = False, single_layer: bool = False)
tgs/models/transformers.py:574
Method__init__
(self, embedding_dim: int, num_embeddings: int)
tgs/models/transformers.py:605
Method__init__
( self, embedding_dim: int, out_dim: int, num_groups: int, act_fn: Opt
tgs/models/transformers.py:643
Method__init__
(self, in_channels: int = 80, sh_degree: int = 3, init_scalin
tgs/models/networks.py:108
Method__init__
(self, size_in, size_out=None, size_h=None)
tgs/models/networks.py:171
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:78
Method__init__
(self, config)
tgs/models/tokenizers/dinov2.py:175
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:215
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:320
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:336
Method__init__
(self, config)
tgs/models/tokenizers/dinov2.py:384
Method__init__
(self, drop_prob: Optional[float] = None)
tgs/models/tokenizers/dinov2.py:425
Method__init__
(self, config)
tgs/models/tokenizers/dinov2.py:456
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:475
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:552
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:733
Method__init__
(self, config: Dinov2Config)
tgs/models/tokenizers/dinov2.py:838
Method__init__
(self, config)
tgs/models/tokenizers/dinov2.py:942
Method__init__
(self, image_size: int, patch_size: int, num_channels: int, hidden_size: int)
tgs/models/tokenizers/dinov2.py:1056
Method__init__
(self, image_size: int, patch_size: int, num_channels: int, hidden_size: int)
tgs/models/tokenizers/dinov2.py:1097
Method__init__
Snowflake Point Deconvolution
tgs/models/snowflake/SPD_pp.py:8
Method__init__
(self, input_channels=1152, dim_feat=512, num_p0=512, radius=1, bounding=True, up_factors=Non
tgs/models/snowflake/model_spdpp.py:79
Method__init__
(self, in_channel, out_channel, kernel_size=1, stride=1, if_bn=True, activation_fn=torch.relu)
tgs/models/snowflake/utils.py:13
Method__init__
(self, in_channel, out_channel, kernel_size=(1, 1), stride=(1, 1), if_bn=True, activation_fn=torch.relu)
tgs/models/snowflake/utils.py:31
Method__init__
(self, in_channel, layer_dims, bn=None)
tgs/models/snowflake/utils.py:66
Method__init__
(self, in_dim=128, hidden_dim=None, out_dim=128)
tgs/models/snowflake/utils.py:83
Method__init__
Args: npoint: int, number of points to sample nsample: int, number of points in each local region radius:
tgs/models/snowflake/utils.py:168
Method__init__
Args: in_channel: int, input channel of points2 mlp: list of int use_points1: boolean, if use points
tgs/models/snowflake/utils.py:217
Method__init__
Args: npoint: int, number of points to sample nsample: int, number of points in each local region radius:
tgs/models/snowflake/utils.py:335
Method__init__
(self, in_channel, dim=256, n_knn=16, pos_hidden_dim=64, attn_hidden_multiplier=4)
tgs/models/snowflake/utils.py:400
Method__init__
(self, d, intermediate_dim, swap=False)
tgs/models/snowflake/utils.py:464
Method__init__
(self, layersList)
tgs/models/snowflake/utils.py:509
Method__init__
(self, name='weight', dim=0, eps=1e-12)
tgs/models/snowflake/utils.py:545
Method__init__
(self, in_channel, dim=256, n_knn=16, pos_hidden_dim=64, attn_hidden_multiplier=4)
tgs/models/snowflake/skip_transformer.py:10
Method__init__
(self, *, device: torch.device, dtype: torch.dtype, heads: int, n_data: int)
tgs/models/snowflake/attention.py:83
Method__init__
(self, *, device: torch.device, dtype: torch.dtype, heads: int, n_ctx: int)
tgs/models/snowflake/attention.py:108
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_data: int,
tgs/models/snowflake/attention.py:131
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int,
tgs/models/snowflake/attention.py:167
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_data: int,
tgs/models/snowflake/attention.py:195
Method__init__
Snowflake Point Deconvolution
tgs/models/snowflake/SPD_crossattn.py:11
Method__init__
Snowflake Point Deconvolution
tgs/models/snowflake/SPD.py:11
Method__init__
(self, radius, nsample, use_xyz=True)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:291
Method__init__
(self)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py:23
Method__init__
(self, npoint, radii, nsamples, mlps, bn=True, use_xyz=True)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py:94
Method__init__
(self, mlp, bn=True)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py:160
Method__len__
(self)
tgs/data.py:210
Method_init_weights
Initialize the weights
tgs/models/tokenizers/dinov2.py:629
Method_prune_heads
Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel
tgs/models/tokenizers/dinov2.py:748
Functionadd_spectral_norm
Applies spectral norm to all modules within the scope of a CNF.
tgs/models/snowflake/utils.py:687
Functionapply_spectral_norm
(module)
tgs/models/snowflake/utils.py:690
Functionassert_input_image
(input_image)
gradio_app.py:61
Methodbackward
(ctx, g)
tgs/utils/ops.py:45
Methodbackward
(ctx, *output_grads)
tgs/models/snowflake/attention.py:43
Methodbackward
(ctx, grad_out)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:61
Methodbackward
(ctx, grad_out)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:93
Methodbackward
(ctx, grad_dist, grad_idx)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:132
Methodbackward
r""" Parameters ---------- grad_out : torch.Tensor (B, c, n) tensor with gradients of ouputs Returns
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:164
Methodbackward
r""" Parameters ---------- grad_out : torch.Tensor (B, C, npoint, nsample) tensor of the gradients of the output
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:217
Methodbackward
(ctx, grad_out)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:272
Methodblend_rgba
(image)
tgs/utils/saving.py:146
Functionbuild_latent_flow
(args)
tgs/models/snowflake/utils.py:530
Methodcollate
(self, batch)
tgs/data.py:262
Functionconfig_to_primitive
(config, resolve: bool = True)
tgs/utils/config.py:63
Methodconfigure
(self, *args, **kwargs)
tgs/models/renderer.py:142
Methodconfigure
(self, *args, **kwargs)
tgs/models/renderer.py:205
Methodconfigure
(self)
tgs/models/transformers.py:721
Methodconfigure
(self)
tgs/models/networks.py:38
Methodconfigure
(self)
tgs/models/tokenizers/point.py:15
Methodconfigure
(self)
tgs/models/tokenizers/image.py:30
Methodconfigure
(self)
tgs/models/tokenizers/triplane.py:20
Methodconfigure
(self)
tgs/models/pointclouds/simplepoint.py:34
Methodconfigure
(self)
tgs/models/pointclouds/pointnet.py:37
Methodconfigure
(self)
tgs/models/snowflake/model_spdpp.py:177
Methodcreate_custom_forward
(module)
tgs/models/tokenizers/dinov2.py:580
Methodcustom_forward
(*inputs)
tgs/models/tokenizers/dinov2.py:581
Methoddetokenize
( self, tokens: Float[Tensor, "B Ct Nt"] )
tgs/models/tokenizers/point.py:26
Methoddetokenize
( self, tokens: Float[Tensor, "B Ct Nt"] )
tgs/models/tokenizers/triplane.py:40
Methoddisable_xformers_memory_efficient_attention
r""" Disable memory efficient attention from [xFormers](https://facebookresearch.github.io/xformers/).
tgs/models/transformers.py:67
Functiondump_config
(path: str, config)
tgs/utils/config.py:67
← previousnext →101–200 of 315, ranked by callers