MCPcopy Create free account

hub / github.com/TUM-AVS/target-bench / functions

Functions13,018 in github.com/TUM-AVS/target-bench

↓ 7 callersMethodredux
models/vipe/csrc/include/eigen3/Eigen/src/Core/Redux.h:408
↓ 7 callersMethodreverse
models/vipe/csrc/include/eigen3/Eigen/src/Core/Reverse.h:119
↓ 7 callersFunctionrows
models/vipe/csrc/include/eigen3/Eigen/src/Core/CwiseNullaryOp.h:60
↓ 7 callersMethodrows
models/vipe/csrc/include/eigen3/Eigen/src/QR/ColPivHouseholderQR.h:327
↓ 7 callersMethodrowwise
models/vipe/csrc/include/eigen3/Eigen/src/Core/VectorwiseOp.h:776
↓ 7 callersFunctionsample_features5d
r"""Sample spatio-temporal features `sample_features5d(input, coords)` works in the same way as :func:`sample_features4d` but for spatio-temp
models/spatracker/models/SpaTrackV2/utils/model_utils.py:390
↓ 7 callersFunctionscaled_index_add
( input: torch.Tensor, # [B, M, D] index: torch.Tensor, # [Bi] - int64 source: torch.Tensor, #
models/vipe/vipe/ext/xformers.py:38
↓ 7 callersMethodseek
(self, offset: int, whence: int = 0)
models/spatracker/models/moge/utils/webfile.py:38
↓ 7 callersMethodsetStorageType
models/vipe/csrc/include/eigen3/Eigen/src/SuperLUSupport/SuperLUSupport.h:145
↓ 7 callersFunctionsinusoidal_embedding_1d
(dim, position)
DiffSynth-Studio/diffsynth/models/wan_video_dit.py:68
↓ 7 callersMethodsolveInPlace
models/vipe/csrc/include/eigen3/Eigen/src/SparseCore/TriangularSolver.h:178
↓ 7 callersMethodstartCol
models/vipe/csrc/include/eigen3/Eigen/src/SparseCore/SparseBlock.h:272
↓ 7 callersMethodstartVec
\internal * \sa insertBack, insertBackByOuterInner */
models/vipe/csrc/include/eigen3/Eigen/src/SparseCore/SparseMatrix.h:415
↓ 7 callersMethodtile
(self, model_input, tile_size, tile_stride, tile_device, tile_dtype)
DiffSynth-Studio/diffsynth/models/tiler.py:20
↓ 7 callersMethodto
(self, *args, **kwargs)
DiffSynth-Studio/diffsynth/trainers/utils.py:371
↓ 7 callersMethodtoRotationMatrix
models/vipe/csrc/include/eigen3/Eigen/src/Geometry/AngleAxis.h:217
↓ 7 callersMethodtoRotationMatrix
models/vipe/csrc/include/eigen3/Eigen/src/Geometry/RotationBase.h:45
↓ 7 callersMethodtranspose_for_scores
(self, x)
DiffSynth-Studio/diffsynth/extensions/ImageQualityMetric/BLIP/med.py:130
↓ 7 callersMethodupsample
(self, x, scale_factor=2)
models/vipe/vipe/priors/depth/metric3d/model/decode_heads/HourGlassDecoder.py:252
↓ 7 callersMethodvaluePtr
* Return the array of nonzero values packed by column * * The size is nnz */
models/vipe/csrc/include/eigen3/Eigen/src/SparseLU/SparseLU_SupernodalMatrix.h:90
↓ 7 callersMethodvec
(self)
models/vipe/vipe/ext/lietorch/groups.py:75
↓ 7 callersFunctionweighted_mean
(x: torch.Tensor, w: torch.Tensor = None, dim: Union[int, torch.Size] = None, keepdim: bool = False, eps: floa
models/spatracker/models/SpaTrackV2/models/tracker3D/spatrack_modules/geometry_torch.py:16
↓ 6 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: int = 3, str
DiffSynth-Studio/diffsynth/models/cog_vae.py:58
↓ 6 callersMethod__init__
(self, config)
DiffSynth-Studio/diffsynth/models/nexus_gen_projector.py:182
↓ 6 callersMethod__init__
( self, dim: int, in_dim: int, ffn_dim: int, out_dim: int, tex
DiffSynth-Studio/diffsynth/models/wan_video_dit_s2v.py:362
↓ 6 callersMethod__init__
(self, add_positional_conv=None)
DiffSynth-Studio/diffsynth/models/svd_unet.py:246
↓ 6 callersMethod__init__
( self, patch_size=(2, 2), in_chans=4, embed_dim=1408, bias=True,
DiffSynth-Studio/diffsynth/models/hunyuan_dit.py:177
↓ 6 callersMethod__init__
(self, h_planes=128, i_planes=128)
models/vipe/vipe/slam/networks/droid_net.py:374
↓ 6 callersMethod__init__
( self, input_dim: int, expansion: int = 4, dropout: float = 0.0, gate
models/vipe/vipe/priors/depth/unidepth/layers.py:40
↓ 6 callersMethod__init__
(self, params)
models/vipe/vipe/priors/depth/unidepth/utils/camera.py:978
↓ 6 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
models/vipe/vipe/priors/depth/unidepth/models/encoder.py:372
↓ 6 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
models/vipe/vipe/priors/track_anything/groundingdino/models/main/backbone/swin_transformer.py:27
↓ 6 callersMethod__init__
(self, indim, outdim, kernel_size, gn_groups=8)
models/vipe/vipe/priors/track_anything/aot/networks/layers/basic.py:68
↓ 6 callersFunction_to_tuple
(x, dim=2)
DiffSynth-Studio/diffsynth/models/hunyuan_video_dit.py:11
↓ 6 callersMethodadd_factors
add edges to factor graph
models/vipe/vipe/slam/components/factor_graph.py:120
↓ 6 callersFunctionadd_general_parsers
(parser)
DiffSynth-Studio/diffsynth/trainers/text_to_image.py:119
↓ 6 callersMethodaffine
models/vipe/csrc/include/eigen3/Eigen/src/Geometry/Transform.h:399
↓ 6 callersFunctionaligned_free
\internal Frees memory allocated with aligned_malloc. */
models/vipe/csrc/include/eigen3/Eigen/src/Core/util/Memory.h:198
↓ 6 callersFunctionangle_diff_vec3
(v1: torch.Tensor, v2: torch.Tensor, eps: float = 1e-12)
models/spatracker/models/moge/utils/geometry_torch.py:72
↓ 6 callersMethodassignCoeff
models/vipe/csrc/include/eigen3/Eigen/src/Core/TriangularMatrix.h:776
↓ 6 callersMethodassign_identity
(self, one_hot_mask)
models/vipe/vipe/priors/track_anything/aot/networks/engines/aot_engine.py:174
↓ 6 callersFunctionattention
(q, k, v)
DiffSynth-Studio/diffsynth/models/hunyuan_video_dit.py:387
↓ 6 callersFunctionbalanced_binary_cross_entropy
logits: Tensor of arbitrary shape targets: same shape as logits balance_weight: scaling the loss reduction: 'mean', 'sum', or 'none'
models/spatracker/models/SpaTrackV2/models/tracker3D/co_tracker/utils.py:653
↓ 6 callersFunctionbitwise_andnot
models/vipe/csrc/include/eigen3/Eigen/src/Core/arch/GPU/PacketMath.h:156
↓ 6 callersFunctionbroadcast
models/vipe/csrc/scatter_ext/scatter.cpp:19
↓ 6 callersMethodbuild_remapping_table
(self, frames_guide, frames_style, patch_match_engine, batch_size, desc="")
DiffSynth-Studio/diffsynth/extensions/FastBlend/runners/fast.py:34
↓ 6 callersMethodcache
(self, desc: str = "Caching", online: bool = False)
models/vipe/vipe/streams/base.py:467
↓ 6 callersMethodclear_cache
(self)
DiffSynth-Studio/diffsynth/models/wan_video_vae.py:1048
↓ 6 callersFunctioncolIndexByOuterInner
models/vipe/csrc/include/eigen3/Eigen/src/Core/AssignEvaluator.h:701
↓ 6 callersMethodcols
models/vipe/csrc/include/eigen3/Eigen/src/Core/Reshaped.h:178
↓ 6 callersMethodcols
models/vipe/csrc/include/eigen3/Eigen/src/PardisoSupport/PardisoSupport.h:139
↓ 6 callersMethodcolsPermutation
Get the permutation that was applied to columns of A
models/vipe/csrc/include/eigen3/Eigen/src/SPQRSupport/SuiteSparseQRSupport.h:216
↓ 6 callersMethodcolwise
models/vipe/csrc/include/eigen3/Eigen/src/Core/VectorwiseOp.h:762
↓ 6 callersMethodcompute
* Compute the symbolic and numeric factorization of the input sparse matrix. * The input matrix should be in column-major storage. */
models/vipe/csrc/include/eigen3/Eigen/src/SparseLU/SparseLU.h:182
↓ 6 callersFunctioncompute_all_metrics
Compute all trajectory metrics at once. Metrics computed: - ADE: Average Displacement Error - FDE: Final Displacement Error
evaluation/metrics.py:318
↓ 6 callersFunctioncompute_overall_score
Compute Overall Trajectory Evaluation Score as defined in metrics.md. The overall score is a weighted combination of normalized metrics:
evaluation/metrics.py:266
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
models/spatracker/models/monoD/zoeDepth/midas_c/midas/transforms.py:94
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
models/spatracker/models/monoD/zoeDepth/models/base_models/midas.py:100
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
models/spatracker/models/monoD/depth_anything_v2/util/transform.py:51
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
models/spatracker/models/monoD/depth_anything/util/transform.py:100
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
models/vipe/vipe/priors/depth/dav2/util/transform.py:54
↓ 6 callersMethoddecode_video
(self, latents, tiled=False, tile_size=64, tile_stride=32)
DiffSynth-Studio/diffsynth/pipelines/sd_video.py:125
↓ 6 callersFunctiondepth2disparity
(depth, return_mask=False)
models/vipe/vipe/priors/depth/priorda/utils.py:36
↓ 6 callersFunctiondepth_to_points_colmap
Unproject a depth map to a point cloud in COLMAP convention. Args: metric_depth: (B, H, W) depth map, meters. intrinsics:
models/spatracker/models/SpaTrackV2/models/tracker3D/spatrack_modules/utils.py:673
↓ 6 callersFunctioneigen_neon_shuffle_mask
models/vipe/csrc/include/eigen3/Eigen/src/Core/arch/NEON/PacketMath.h:111
↓ 6 callersFunctioneq_mask
models/vipe/csrc/include/eigen3/Eigen/src/Core/arch/GPU/PacketMath.h:165
↓ 6 callersMethodestimate_nnf
(self, source_guide, target_guide, source_style)
DiffSynth-Studio/diffsynth/extensions/FastBlend/patch_match.py:283
↓ 6 callersMethodevalTo
models/vipe/csrc/include/eigen3/Eigen/src/SparseQR/SparseQR.h:625
↓ 6 callersFunctioneval_expr_given_size
Replace symbolic last/end "keywords" by their true runtime value
models/vipe/csrc/include/eigen3/Eigen/src/Core/util/IndexedViewHelper.h:63
↓ 6 callersFunctionexample
(pipe, seeds, example_id, global_prompt, entity_prompts)
DiffSynth-Studio/examples/qwen_image/model_inference/Qwen-Image-EliGen.py:65
↓ 6 callersFunctionexample
(pipe, seeds, example_id, global_prompt, entity_prompts)
DiffSynth-Studio/examples/qwen_image/model_inference_low_vram/Qwen-Image-EliGen.py:65
↓ 6 callersFunctionextract_frames
Extract evenly spaced frames from a video. Args: video_path: Path to the video file output_dir: Directory to save extrac
pipelines/vggt/scale_factor/compute_scale_factor.py:785
↓ 6 callersMethodfps
(self)
models/vipe/vipe/streams/base.py:277
↓ 6 callersMethodfreeze_parameters
(self)
DiffSynth-Studio/diffsynth/trainers/text_to_image.py:30
↓ 6 callersMethodget
models/vipe/csrc/include/eigen3/Eigen/src/SparseCore/SparseDenseProduct.h:269
↓ 6 callersFunctionget_activation
(name, bank)
models/spatracker/models/monoD/zoeDepth/models/base_models/midas.py:44
↓ 6 callersMethodget_data
Abstract method to retrieve data for a given sequence. Args: seq_index (int, optional): Index of the sequence
models/vggt/training/data/base_dataset.py:75
↓ 6 callersMethodget_observations
Args: view_idx: The index of the view. frame_idx: The index of the frame. keypoint_indices: The indices o
models/vipe/vipe/slam/components/sparse_tracks/__init__.py:50
↓ 6 callersFunctionget_points_on_a_grid
(grid_size, interp_shape, grid_center=(0, 0), device="cuda")
models/spatracker/models/SpaTrackV2/models/utils.py:1090
↓ 6 callersMethodinnerSize
models/vipe/csrc/include/eigen3/Eigen/src/SparseCore/SparseView.h:61
↓ 6 callersMethodiproj_disp
( self, disps: torch.Tensor, disps_u: torch.Tensor, disps_v: torch.Tensor,
models/vipe/vipe/utils/cameras.py:228
↓ 6 callersFunctionlaunch_training_task
(model, args)
DiffSynth-Studio/diffsynth/trainers/text_to_image.py:267
↓ 6 callersMethodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, **kwargs)
DiffSynth-Studio/diffsynth/models/model_manager.py:143
↓ 6 callersFunctionload_ground_truth_trajectory
Load ground truth trajectory from CSV file. Args: segment_path: Path to the segment directory segment_name: Name of the
pipelines/vggt/extrinsic_path/extract_extrinsic_path.py:159
↓ 6 callersFunctionlt_mask
models/vipe/csrc/include/eigen3/Eigen/src/Core/arch/GPU/PacketMath.h:174
↓ 6 callersMethodmatch
(self, file_path="", state_dict={})
DiffSynth-Studio/diffsynth/models/model_manager.py:140
↓ 6 callersMethodmatrixQ
\returns an expression of the matrix Q as products of sparse Householder reflectors. * The common usage of this function is to apply it to a dense
models/vipe/csrc/include/eigen3/Eigen/src/SparseQR/SparseQR.h:186
↓ 6 callersMethodmatrixU
\returns an expression of the matrix U, * The only operation available with this expression is the triangular solve * \code * y = b;
models/vipe/csrc/include/eigen3/Eigen/src/SparseLU/SparseLU.h:253
↓ 6 callersMethodnonZerosEstimate
models/vipe/csrc/include/eigen3/Eigen/src/SparseCore/SparseVector.h:430
↓ 6 callersFunctionnormalized_view_plane_uv
UV with left-top corner as (-width / diagonal, -height / diagonal) and right-bottom corner as (width / diagonal, height / diagonal)
models/spatracker/models/moge/utils/geometry_torch.py:40
↓ 6 callersFunctionouterStride
\returns the outer-stride of the underlying dense matrix * \sa DenseCoeffsBase::outerStride() */
models/vipe/csrc/include/eigen3/Eigen/src/Core/TriangularMatrix.h:371
↓ 6 callersFunctionpbar
A wrapper around tqdm.tqdm that disables the progress bar if the disable_progress_bar flag is set.
models/vipe/vipe/utils/logging.py:48
↓ 6 callersFunctionpexp_float
models/vipe/csrc/include/eigen3/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h:437
↓ 6 callersFunctionpor
models/vipe/csrc/include/eigen3/Eigen/src/Core/arch/AVX512/PacketMath.h:1512
↓ 6 callersFunctionposenc
Cat x with a positional encoding of x with scales 2^[min_deg, max_deg-1]. Instead of computing [sin(x), cos(x)], we use the trig identity cos
models/spatracker/models/SpaTrackV2/models/tracker3D/delta_utils/blocks.py:804
↓ 6 callersFunctionpredux_half_dowto4
models/vipe/csrc/include/eigen3/Eigen/src/Core/products/GeneralBlockPanelKernel.h:710
↓ 6 callersFunctionpredux_max
models/vipe/csrc/include/eigen3/Eigen/src/Core/GenericPacketMath.h:905
↓ 6 callersFunctionpredux_min
models/vipe/csrc/include/eigen3/Eigen/src/Core/GenericPacketMath.h:890
↓ 6 callersMethodprocess_window_sum
(self, frames_guide, blending_table, patch_match_engine, window_size, batch_size, desc="")
DiffSynth-Studio/diffsynth/extensions/FastBlend/runners/fast.py:77
↓ 6 callersMethodprojection
\returns the projection of a point \a p onto the line \c *this. */
models/vipe/csrc/include/eigen3/Eigen/src/Geometry/ParametrizedLine.h:93
← previousnext →601–700 of 13,018, ranked by callers