MCPcopy Create free account

hub / github.com/Meshcapade/difflocks / functions

Functions981 in github.com/Meshcapade/difflocks

↓ 1 callersFunctioncols
* \brief Returns the number of columns of this matrix. * \returns the number of columns. */
extensions/include/cuMat/src/MatrixBase.h:43
↓ 1 callersFunctioncols
extensions/include/cuMat/src/UnaryOps.h:758
↓ 1 callersMethodcols
extensions/include/cuMat/src/SparseMatrixBase.h:429
↓ 1 callersMethodcols
extensions/include/cuMat/src/MatrixBlock.h:70
↓ 1 callersMethodcompiled_function
(self)
k_diffusion/models/flags.py:43
↓ 1 callersFunctioncompute_loss
(phase, gt_dict, pred_dict, hyperparams)
train_rgb2material.py:99
↓ 1 callersFunctioncompute_loss_curv_l1
(gt_hair_strands, pred_hair_strands)
losses/losses.py:61
↓ 1 callersFunctioncompute_loss_dir_l1
(gt_hair_strands, pred_hair_strands)
losses/losses.py:42
↓ 1 callersFunctioncompute_loss_kl
(mean, logstd)
losses/losses.py:82
↓ 1 callersFunctioncompute_loss_l1
(gt_hair_strands, pred_hair_strands)
losses/losses.py:15
↓ 1 callersMethodcompute_tbn
(self, mesh_verts, mesh_uv, mesh_faces, mesh_v_tangents, mesh_v_bitangents, mesh_v_normals)
data_loader/dataloader.py:65
↓ 1 callersFunctioncompute_uv_space_data
Compute a UV-space barycentric map where each texel contains barycentric coordinates for the closest point on a UV triangle. Args: vt:
data_loader/mesh_utils.py:186
↓ 1 callersFunctioncompute_vertex_tbn
Compute tangents, bitangents, normals. Args: vertex_pos: [N,3] vertex coordinates vertex_uv: [N,2] texture coordinates fac
data_loader/mesh_utils.py:51
↓ 1 callersMethodconjugate
extensions/include/cuMat/src/TransposeOp.h:178
↓ 1 callersFunctioncopyToHost
* \brief Performs a synchronous copy from the * device memory of this matrix into the * specified host memory * This copy is synchronized on the de
extensions/include/cuMat/src/Matrix.h:847
↓ 1 callersFunctioncreate_dataloaders
(dataset_path, dataset_processed_path)
train_rgb2material.py:47
↓ 1 callersFunctioncreate_dataloaders
(dataset_path)
train_strandsVAE.py:67
↓ 1 callersFunctioncrop_face
(image, face_landmarks, output_size, crop_size_multiplier=2.8)
inference/img2hair.py:129
↓ 1 callersFunctioncubic_spline_coeffs
(t, x)
utils/strand_util.py:299
↓ 1 callersMethodcublasGeam
extensions/include/cuMat/src/CublasApi.h:159
↓ 1 callersMethodcublasGemm
extensions/include/cuMat/src/CublasApi.h:196
↓ 1 callersMethodcublasGemmBatched
extensions/include/cuMat/src/CublasApi.h:275
↓ 1 callersMethodcusolverGetrf
extensions/include/cuMat/src/CusolverApi.h:218
↓ 1 callersMethodcusolverGetrs
extensions/include/cuMat/src/CusolverApi.h:249
↓ 1 callersMethodcusolverPotrf
extensions/include/cuMat/src/CusolverApi.h:151
↓ 1 callersMethodcusolverPotrs
extensions/include/cuMat/src/CusolverApi.h:181
↓ 1 callersFunctiondct
(x)
k_diffusion/layers.py:16
↓ 1 callersFunctiondeepClone
extensions/include/cuMat/src/Matrix.h:1220
↓ 1 callersFunctiondeepCloneImpl
extensions/include/cuMat/src/Matrix.h:1191
↓ 1 callersFunctiondeepCloneImpl_directTranspose
extensions/include/cuMat/src/Matrix.h:1163
↓ 1 callersFunctiondirectTranspose
extensions/include/cuMat/src/TransposeOp.h:66
↓ 1 callersFunctiondownscale_pos
(pos)
k_diffusion/models/modules.py:56
↓ 1 callersMethoddpm_solver_adaptive
(self, x, t_start, t_end, order=3, rtol=0.05, atol=0.0078, h_init=0.05, pcoeff=0., icoeff=1., dcoeff=0., accep
k_diffusion/sampling.py:427
↓ 1 callersMethoddpm_solver_fast
(self, x, t_start, t_end, nfe, eta=0., s_noise=1., noise_sampler=None)
k_diffusion/sampling.py:388
↓ 1 callersFunctionenable_stratified
A context manager that enables stratified sampling.
k_diffusion/utils.py:284
↓ 1 callersFunctionensure_distributed
()
train_scalp_diffusion.py:44
↓ 1 callersFunctionexport_alembic
(out_alembic_path, resolution)
inference/npz2blender.py:35
↓ 1 callersMethodextra_repr
(self)
k_diffusion/models/modules.py:171
↓ 1 callersMethodforward
(self, x, *args, **kwargs)
k_diffusion/models/modules.py:556
↓ 1 callersFunctionfreq_weight_1d
(n, scales=0, dtype=None, device=None)
k_diffusion/layers.py:27
↓ 1 callersFunctionfreq_weight_nd
(shape, scales=0, dtype=None, device=None)
k_diffusion/layers.py:36
↓ 1 callersFunctionfreqs_pixel_log
(max_freq=10.0)
k_diffusion/models/axial_rope.py:78
↓ 1 callersFunctionfw_ceil
(x, fw)
utils/resize_right/resize_right.py:358
↓ 1 callersFunctionfw_conv
(input, filter, stride)
utils/resize_right/resize_right.py:403
↓ 1 callersFunctionfw_empty
(shape, fw, device)
utils/resize_right/resize_right.py:424
↓ 1 callersFunctiongetRawCoeff
extensions/include/cuMat/src/Matrix.h:685
↓ 1 callersFunctionget_cli_args
()
train_scalp_diffusion.py:48
↓ 1 callersFunctionget_field_of_view
(projected_grid, cur_support_sz, fw, eps, device)
utils/resize_right/resize_right.py:141
↓ 1 callersMethodget_freqs
(self, pos)
k_diffusion/models/axial_rope.py:99
↓ 1 callersFunctionget_offset
(beta2, rho_inf)
schedulers/pytorch_warmup/radam.py:17
↓ 1 callersFunctionget_projected_grid
(in_sz, out_sz, scale_factor, fw, by_convs, device=None)
utils/resize_right/resize_right.py:126
↓ 1 callersMethodget_scalp
(self)
data_loader/dataloader.py:646
↓ 1 callersFunctionget_strand_length
(strand)
utils/strand_util.py:148
↓ 1 callersFunctionget_strands
(points)
utils/strand_util.py:89
↓ 1 callersFunctionget_use_compile
()
k_diffusion/models/flags.py:9
↓ 1 callersMethodget_value
Gets the current EMA decay rate.
k_diffusion/utils.py:145
↓ 1 callersFunctionget_weights
(interp_method, projected_grid, field_of_view)
utils/resize_right/resize_right.py:204
↓ 1 callersFunctionhorizontally_flip
(batch)
data_processing/create_latents.py:28
↓ 1 callersFunctionhorizontally_flip
(batch, scalp_mesh_data)
data_processing/create_scalp_textures.py:138
↓ 1 callersMethodinnerSize
extensions/include/cuMat/src/ProductOp.h:221
↓ 1 callersFunctioninplace
* \brief Forces inplace assignment. * The only usecase is <code>matrix.inplace() = expression</code> * where the content's of matrix are ove
extensions/include/cuMat/src/Matrix.h:1136
↓ 1 callersFunctioninverse_FDT
(code)
utils/general_util.py:117
↓ 1 callersMethodisInitialized
* \brief Checks if this matrix is initialized with a sparsity pattern and can therefore be used in expressions. */
extensions/include/cuMat/src/SparseMatrixBase.h:414
↓ 1 callersFunctionkl
(mean, logstd)
losses/losses.py:96
↓ 1 callersMethodlimiter
(self, x)
k_diffusion/sampling.py:315
↓ 1 callersFunctionlinear_geglu
(x, weight, bias=None)
k_diffusion/models/modules.py:94
↓ 1 callersFunctionlinear_multistep_coeff
(order, t, i, j)
k_diffusion/sampling.py:247
↓ 1 callersMethodlogDeterminant
* \brief Computes the log-determinant of this matrix. * This is only supported for hermitian positive definite matrices, because no sign is comput
extensions/include/cuMat/src/LUDecomposition.h:174
↓ 1 callersFunctionlogsnr_schedule_cosine
(t, logsnr_min, logsnr_max)
k_diffusion/utils.py:358
↓ 1 callersFunctionlogsnr_schedule_cosine_interpolated
(t, image_d, noise_d_low, noise_d_high, logsnr_min, logsnr_max)
k_diffusion/utils.py:367
↓ 1 callersMethodloss
(self, input, noise, sigma, **kwargs)
k_diffusion/layers.py:89
↓ 1 callersFunctionmain
()
train_rgb2material.py:268
↓ 1 callersFunctionmain
()
train_strandsVAE.py:258
↓ 1 callersFunctionmain
()
train_scalp_diffusion.py:98
↓ 1 callersFunctionmain
()
utils/create_strand_latent_weights.py:76
↓ 1 callersFunctionmain
()
inference/npz2blender.py:109
↓ 1 callersFunctionmain
()
data_processing/uncompress_data.py:17
↓ 1 callersFunctionmain
()
data_processing/create_latents.py:78
↓ 1 callersFunctionmain
()
data_processing/create_scalp_textures.py:181
↓ 1 callersFunctionmain
()
data_processing/create_chunked_strands.py:20
↓ 1 callersFunctionmakeExclusiveUse
* \brief Checks if the underlying data is used by an other matrix expression and if so, * copies the data so that this matrix is the exclusive use
extensions/include/cuMat/src/SparseMatrix.h:269
↓ 1 callersFunctionmake_cfg_model_fn
(model)
train_scalp_diffusion.py:288
↓ 1 callersFunctionmake_grid
(h_pos, w_pos)
k_diffusion/models/axial_rope.py:36
↓ 1 callersFunctionmake_shifted_window_masks
(n_h_w, n_w_w, w_h, w_w, shift, device=None)
k_diffusion/models/modules.py:300
↓ 1 callersMethodmallocDevice
* \brief Allocates size-number of bytes on the device system. * This memory must be freed with freeDevice(void*). * * Note that allocate zero bytes
extensions/include/cuMat/src/Context.h:271
↓ 1 callersFunctionmaxIterations
\returns the max number of iterations. * It is either the value setted by setMaxIterations or, by default, * twice the number of columns o
extensions/include/cuMat/src/IterativeSolverBase.h:83
↓ 1 callersFunctionnatural_cubic_spline_coeffs
Calculates the coefficients of the natural cubic spline approximation to the batch of controls given. Arguments: t: Tensor of times.
utils/strand_util.py:189
↓ 1 callersFunctionnormalize_data_2D
(strands_data, mean, std)
models/strand_codec.py:29
↓ 1 callersMethodparam_groups
(self, base_lr=5e-4, mapping_lr_scale=1 / 3)
k_diffusion/models/image_transformer_v2_conditional.py:205
↓ 1 callersFunctionprepare_gt_batch
(batch, hyperparams, do_augmentation=False)
train_rgb2material.py:143
↓ 1 callersFunctionprepare_gt_batch
(batch, hyperparams, world2local, do_augmentation=False)
train_strandsVAE.py:98
↓ 1 callersFunctionprepare_gt_batch
(batch)
data_processing/create_scalp_textures.py:40
↓ 1 callersMethodpropose_step
(self, error)
k_diffusion/sampling.py:318
↓ 1 callersFunctionrepr
(model)
utils/general_util.py:194
↓ 1 callersMethodrgb2hair
(self, rgb_img, out_path=None)
inference/img2hair.py:275
↓ 1 callersFunctionrho_inf_fn
(beta2)
schedulers/pytorch_warmup/radam.py:7
↓ 1 callersFunctionrotate_half
(x)
k_diffusion/models/axial_rope.py:13
↓ 1 callersFunctionround_to_power_of_two
(x, tol)
k_diffusion/config.py:11
↓ 1 callersFunctionrows
* \brief Returns the number of rows of this matrix. * \returns the number of rows. */
extensions/include/cuMat/src/MatrixBase.h:38
↓ 1 callersFunctionrows
extensions/include/cuMat/src/UnaryOps.h:751
← previousnext →201–300 of 981, ranked by callers