MCPcopy Create free account

hub / github.com/Meshcapade/difflocks / functions

Functions981 in github.com/Meshcapade/difflocks

↓ 3 callersFunctionfw_arange
(upper_bound, fw, device)
utils/resize_right/resize_right.py:417
↓ 3 callersMethodget_scalings
(self, sigma)
k_diffusion/layers.py:83
↓ 3 callersFunctionimg_2_pca
(img)
utils/vis_util.py:32
↓ 3 callersFunctioninit
(shape)
k_diffusion/models/axial_rope.py:72
↓ 3 callersFunctioninterpolate_tbn
(barys, vertex_idxs, v_tangents, v_bitangents, v_normals)
data_loader/mesh_utils.py:112
↓ 3 callersFunctionisExclusiveUse
* \brief Checks if the this matrix has exclusive use to the underlying data, i.e. no other matrix expression shares the data. * This allows to ch
extensions/include/cuMat/src/Matrix.h:746
↓ 3 callersFunctionload
extensions/include/cuMat/src/CudaUtils.h:20
↓ 3 callersFunctionnormalize_data_3D
(strands_data, mean, std)
models/strand_codec.py:11
↓ 3 callersMethodouterSize
extensions/include/cuMat/src/SparseMatrixBase.h:453
↓ 3 callersMethodrun
(self, rgb_image_numpy)
inference/img2hair.py:68
↓ 3 callersFunctionshifted_window
(window_size, window_shift, x)
k_diffusion/models/modules.py:287
↓ 3 callersMethodsolve
extensions/include/cuMat/src/ConjugateGradient.h:241
↓ 3 callersFunctiontag_module
(module, tag)
k_diffusion/models/modules.py:71
↓ 3 callersMethodto_dict
(self)
data_loader/dataloader.py:78
↓ 3 callersFunctionworld_to_tbn_space
(strands_tbn, strands_positions, root_normals)
data_loader/mesh_utils.py:141
↓ 2 callersMethod__init__
(self, optimizer, warmup_params, last_step=-1)
schedulers/pytorch_warmup/base.py:23
↓ 2 callersMethod__init__
(self, do_vae=True, scale_init=30.0, decode_type="dir", decode_random_verts=False, nr_verts_per_strand=256, nr
models/strand_codec.py:381
↓ 2 callersFunction_apply_rotary_emb_inplace
(x, theta, conj)
k_diffusion/models/modules.py:211
↓ 2 callersMethodafter_backward_pass
(self, **kwargs)
callbacks/callback.py:46
↓ 2 callersMethodafter_forward_pass
(self, **kwargs)
callbacks/callback.py:42
↓ 2 callersMethodbatches
extensions/include/cuMat/src/BinaryOps.h:141
↓ 2 callersMethodbatches
extensions/include/cuMat/src/SolverBase.h:109
↓ 2 callersMethodbefore_backward_pass
(self, **kwargs)
callbacks/callback.py:44
↓ 2 callersMethodbefore_forward_pass
(self, **kwargs)
callbacks/callback.py:40
↓ 2 callersFunctioncenters
(start, stop, num, dtype=None, device=None)
k_diffusion/models/axial_rope.py:31
↓ 2 callersMethodcoeff
extensions/include/cuMat/src/BinaryOps.h:163
↓ 2 callersMethodcols
extensions/include/cuMat/src/BinaryOps.h:137
↓ 2 callersMethodcols
extensions/include/cuMat/src/CholeskyDecomposition.h:114
↓ 2 callersMethodcols
extensions/include/cuMat/src/SolverBase.h:108
↓ 2 callersFunctioncompute_curv
(strand_dirs, append_last_curv=True)
utils/strand_util.py:520
↓ 2 callersMethodcompute_scalp_data
(scalp_path)
data_loader/dataloader.py:679
↓ 2 callersFunctioncreate_callbacks
(with_tensorboard, with_visualizer, experiment_name, viewer_config_path=None)
callbacks/callback_utils.py:8
↓ 2 callersMethoddampen
Dampen the learning rates. Arguments: step (int): The index of current step. (Default: None)
schedulers/pytorch_warmup/base.py:47
↓ 2 callersMethoddampening
(self)
schedulers/pytorch_warmup/base.py:62
↓ 2 callersMethoddeterminant
* \brief Computes the determinant of this matrix * \return The determinant */
extensions/include/cuMat/src/LUDecomposition.h:154
↓ 2 callersMethoddpm_solver_1_step
(self, x, t, t_next, eps_cache=None)
k_diffusion/sampling.py:358
↓ 2 callersMethoddpm_solver_3_step
(self, x, t, t_next, r1=1 / 3, r2=2 / 3, eps_cache=None)
k_diffusion/sampling.py:375
↓ 2 callersMethodepoch_ended
(self, **kwargs)
callbacks/callback.py:34
↓ 2 callersMethodepoch_started
(self, **kwargs)
callbacks/callback.py:28
↓ 2 callersMethodevaluate
(self, t)
utils/strand_util.py:271
↓ 2 callersMethodfile2hair
(self, file_path, out_path)
inference/img2hair.py:388
↓ 2 callersMethodfreeDevice
* \brief Frees memory previously allocated with allocateDevice(size_t). * Passing a NULL-pointer should be a no-op. * \param memory the memory to be
extensions/include/cuMat/src/Context.h:311
↓ 2 callersFunctionfw_pad
(x, fw, pad_sz, pad_mode, dim=0)
utils/resize_right/resize_right.py:386
↓ 2 callersFunctiongenerate_latents_dinov2
(args, batch, preprocessor, model, output_latents_path)
data_processing/create_latents.py:37
↓ 2 callersFunctiongenerate_scalp_textures
(batch, model, normalization_dict, tex_size, output_scalp_texture_path)
data_processing/create_scalp_textures.py:66
↓ 2 callersMethodgetNNZ
extensions/include/cuMat/src/SparseMatrixBase.h:124
↓ 2 callersFunctionget_warmup_params
(warmup_period, group_count)
schedulers/pytorch_warmup/base.py:73
↓ 2 callersFunctionget_window
Return a window function. Args: win_type (str): Window type. Can either be one of the window function provided in PyTorch ['h
utils/general_util.py:271
↓ 2 callersFunctionisExclusiveUse
* \brief Checks if the this matrix has exclusive use to the underlying data, i.e. no other matrix expression shares the data. * \b Note: The data
extensions/include/cuMat/src/SparseMatrix.h:255
↓ 2 callersMethodload_chunked_paths
(self,chunk=100)
data_loader/dataloader.py:112
↓ 2 callersFunctionlogsnr_schedule_cosine_shifted
(t, image_d, noise_d, logsnr_min, logsnr_max)
k_diffusion/utils.py:363
↓ 2 callersFunctionmake_axial_pos
(h, w, pixel_aspect_ratio=1.0, align_corners=False, dtype=None, device=None)
k_diffusion/models/axial_rope.py:60
↓ 2 callersFunctionmesh_to_data
(mesh)
data_loader/mesh_utils.py:392
↓ 2 callersFunctionnormalize_gt_data
(gt_dict, normalization_dict)
models/strand_codec.py:66
↓ 2 callersMethodphase_ended
(self, **kwargs)
callbacks/callback.py:32
↓ 2 callersMethodphase_started
(self, **kwargs)
callbacks/callback.py:30
↓ 2 callersFunctionquaternion_to_matrix
Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten
utils/general_util.py:476
↓ 2 callersFunctionrandom_quaternions
Generate random quaternions representing rotations, i.e. versors with nonnegative real part. Args: n: Number of quaternions in a
utils/general_util.py:434
↓ 2 callersMethodrecord
* Records an event into this instance after all preceding operations in <code>stream</code> have been completed. */
extensions/include/cuMat/src/Context.h:527
↓ 2 callersFunctionrho_fn
(t, beta2, rho_inf)
schedulers/pytorch_warmup/radam.py:11
↓ 2 callersFunctionrms_norm
(x, scale, eps)
k_diffusion/models/modules.py:103
↓ 2 callersMethodrows
extensions/include/cuMat/src/BinaryOps.h:133
↓ 2 callersMethodrows
extensions/include/cuMat/src/SolverBase.h:107
↓ 2 callersFunctionscale_for_cosine_sim_single
(q, scale, eps)
k_diffusion/models/attention.py:53
↓ 2 callersMethodsort
(a, b)
k_diffusion/sampling.py:83
↓ 2 callersMethodstep
(self, epoch=None, metrics=None)
schedulers/warmup.py:54
↓ 2 callersFunctionstoreMat
extensions/include/cuMat/src/DenseLinAlgOps.h:319
↓ 2 callersMethodstreamWait
* Synchronizes the given stream with this event. * The given stream will wait until the event recorded with \ref record(cudaStream_t) has completed.
extensions/include/cuMat/src/Context.h:536
↓ 2 callersFunctiontag_param
(param, tag)
k_diffusion/models/modules.py:63
↓ 2 callersFunctionto_snake_case
Converts CamelCase string into snake_case.
callbacks/callback.py:6
↓ 2 callersFunctionun_normalize_data
(strands_data, mean, std)
models/strand_codec.py:47
↓ 1 callersFunctionB_spline_interpolate
(X,num)
utils/strand_util.py:142
↓ 1 callersFunctionConjugateGradient
* \brief Initializes the Conjugate Gradient with the specified matrix. * The preconditioner is created with \code Preconditioner(matrix) \endcode
extensions/include/cuMat/src/ConjugateGradient.h:96
↓ 1 callersFunctionDevicePointer
extensions/include/cuMat/src/DevicePointer.h:59
↓ 1 callersFunctionIterativeSolverBase
extensions/include/cuMat/src/IterativeSolverBase.h:36
↓ 1 callersFunctionSparseMatrix
* \brief Initializes this SparseMatrix with the given sparsity pattern and (if not fixed by the template argument) * with the given number of bat
extensions/include/cuMat/src/SparseMatrix.h:95
↓ 1 callersMethod__init__
(self, mode)
inference/img2hair.py:48
↓ 1 callersMethod__init__
(self, optimizer, last_step=-1)
schedulers/pytorch_warmup/untuned.py:16
↓ 1 callersFunction_copysign
Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding
utils/general_util.py:507
↓ 1 callersMethod_get_closed_form_lr
(self)
k_diffusion/utils.py:190
↓ 1 callersMethod_get_closed_form_lr
(self)
k_diffusion/utils.py:231
↓ 1 callersMethod_get_closed_form_lr
(self)
k_diffusion/utils.py:263
↓ 1 callersMethod_solve_impl
extensions/include/cuMat/src/LUDecomposition.h:185
↓ 1 callersFunction_solve_with_guess_impl
extensions/include/cuMat/src/ConjugateGradient.h:124
↓ 1 callersFunctionapply_antialiasing_if_needed
(interp_method, support_sz, scale_factor, antialiasing)
utils/resize_right/resize_right.py:343
↓ 1 callersFunctionapply_convs
(input, scale_factor, in_sz, out_sz, weights, dim, pad_sz, pad_mode, fw)
utils/resize_right/resize_right.py:251
↓ 1 callersFunctionapply_rotary_emb
(freqs, t, start_index=0, scale=1.0)
k_diffusion/models/axial_rope.py:21
↓ 1 callersFunctionapply_weights
(input, field_of_view, weights, dim, n_dims, pad_sz, pad_mode, fw)
utils/resize_right/resize_right.py:217
↓ 1 callersFunctionapply_window_attention
(window_size, window_shift, q, k, v, scale=None)
k_diffusion/models/modules.py:333
↓ 1 callersMethodassertValid
* \brief Checks with assertions that this SparsityPattern is valid. */
extensions/include/cuMat/src/SparseMatrixBase.h:113
↓ 1 callersFunctionbatched_index_select
(t, dim, inds)
utils/general_util.py:19
↓ 1 callersFunctionbatches
* \brief Returns the number of batches of this matrix. * \returns the number of batches. */
extensions/include/cuMat/src/MatrixBase.h:48
↓ 1 callersFunctionbatches
extensions/include/cuMat/src/UnaryOps.h:765
↓ 1 callersMethodbatches
extensions/include/cuMat/src/MatrixBlock.h:76
↓ 1 callersFunctionbounding_box
(h, w, pixel_aspect_ratio=1.0)
k_diffusion/models/axial_rope.py:42
↓ 1 callersFunctioncalc_pad_sz
(in_sz, out_sz, field_of_view, projected_grid, scale_factor, dim_by_convs, fw, device)
utils/resize_right/resize_right.py:153
↓ 1 callersFunctionclassname
(obj)
callbacks/callback.py:12
↓ 1 callersMethodcoeff
extensions/include/cuMat/src/TransposeOp.h:140
↓ 1 callersMethodcoeff
extensions/include/cuMat/src/SparseExpressionOp.h:63
↓ 1 callersMethodcoeff
extensions/include/cuMat/src/MatrixBlock.h:125
← previousnext →101–200 of 981, ranked by callers