MCPcopy Create free account

hub / github.com/Meshcapade/difflocks / functions

Functions981 in github.com/Meshcapade/difflocks

↓ 1 callersMethodrows
extensions/include/cuMat/src/SparseMatrixBase.h:423
↓ 1 callersMethodrows
extensions/include/cuMat/src/MatrixBlock.h:64
↓ 1 callersFunctionrun
()
inference_difflocks.py:19
↓ 1 callersFunctionsample_from_density_map
(density_map, downsample_lvl=None)
utils/strand_util.py:539
↓ 1 callersFunctionsample_images
(nr_images)
train_scalp_diffusion.py:305
↓ 1 callersFunctionsample_images_cfg
(nr_images, cfg_val, cfg_interval, model_ema, model_config, nr_iters=100, extra_args={}, callback=None)
utils/diffusion_utils.py:21
↓ 1 callersFunctionsample_strands_from_scalp_with_density
(scalp_texture, density_map, strand_codec, normalization_dict, scal
utils/strand_util.py:552
↓ 1 callersFunctionsave
()
train_scalp_diffusion.py:326
↓ 1 callersFunctionscale_for_cosine_sim
(q, k, scale, eps)
k_diffusion/models/attention.py:39
↓ 1 callersFunctionscale_for_cosine_sim_qkv
(qkv, scale, eps)
k_diffusion/models/modules.py:122
↓ 1 callersFunctionscale_for_cosine_sim_qkv
(qkv, scale, eps)
k_diffusion/models/attention.py:48
↓ 1 callersFunctionset_scale_and_out_sz
(in_shape, out_shape, scale_factors, by_convs, scale_tolerance, max_numerator, eps, f
utils/resize_right/resize_right.py:280
↓ 1 callersFunctionshifted_unwindow
(window_shift, x)
k_diffusion/models/modules.py:293
↓ 1 callersMethodstep_ReduceLROnPlateau
(self, metrics, epoch=None)
schedulers/warmup.py:40
↓ 1 callersMethodstore
extensions/include/cuMat/src/DenseLinAlgOps.h:40
↓ 1 callersFunctionstratified_uniform
Draws stratified samples from a uniform distribution.
k_diffusion/utils.py:268
↓ 1 callersMethodsubsample_to_nr_strands
(self, nr_strands_to_select)
data_loader/dataloader.py:54
↓ 1 callersFunctiontrain
(args, hyperparams, loader_train, loader_test, experiment_name, output_training_path)
train_rgb2material.py:158
↓ 1 callersFunctiontrain
(args, hyperparams, loader_train, loader_test, experiment_name, output_training_path)
train_strandsVAE.py:147
↓ 1 callersFunctiontrain_mode
A context manager that places a model into training mode and restores the previous mode on exit.
k_diffusion/utils.py:72
↓ 1 callersFunctiontridiagonal_solve
Solves a tridiagonal system Ax = b. The arguments A_upper, A_digonal, A_lower correspond to the three diagonals of A. Letting U = A_upper, D=A_di
utils/general_util.py:24
↓ 1 callersFunctionunwindow
(x)
k_diffusion/models/modules.py:280
↓ 1 callersMethodwarmup_factor
(self, step, **params)
schedulers/pytorch_warmup/base.py:69
↓ 1 callersFunctionwindow
(window_size, x)
k_diffusion/models/modules.py:267
↓ 1 callersFunctionzero_module
Zero out the parameters of a module and return it.
k_diffusion/models/attention.py:30
MethodAsDiagonalOp
extensions/include/cuMat/src/UnaryOps.h:381
MethodBinaryOp
extensions/include/cuMat/src/BinaryOps.h:110
FunctionCSRMVKernel_StaticBatches
extensions/include/cuMat/src/SparseProductEvaluation.h:46
FunctionCUMAT_STRONG_INLINE DeviceMatrix<Scalar, Dims, Flags> loadMat
extensions/include/cuMat/src/DenseLinAlgOps.h:304
FunctionCUMAT_STRONG_INLINE cdouble conjugateCoeff<cdouble, true>
extensions/include/cuMat/src/TransposeOp.h:52
FunctionCUMAT_STRONG_INLINE cfloat conjugateCoeff<cfloat, true>
extensions/include/cuMat/src/TransposeOp.h:51
MethodCastingOp
extensions/include/cuMat/src/UnaryOps.h:300
MethodCholeskyDecomposition
* \brief Uninitialized Cholesky decomposition, * but with the specified matrix dimensions * Call \ref compute() before use. */
extensions/include/cuMat/src/CholeskyDecomposition.h:51
MethodConstantFunctor
extensions/include/cuMat/src/NullaryOps.h:83
MethodContext
extensions/include/cuMat/src/Context.h:146
MethodCublasApi
extensions/include/cuMat/src/CublasApi.h:78
MethodCusolverApi
extensions/include/cuMat/src/CusolverApi.h:76
FunctionCwiseCSCEvaluationKernel
extensions/include/cuMat/src/SparseEvaluation.h:36
FunctionCwiseCSREvaluationKernel
extensions/include/cuMat/src/SparseEvaluation.h:15
FunctionCwiseELLPACKEvaluationKernel
extensions/include/cuMat/src/SparseEvaluation.h:57
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:39
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:78
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:125
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:172
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:220
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:274
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:328
MethodDenseStorage
extensions/include/cuMat/src/Matrix.h:383
FunctionDeterminantKernel
extensions/include/cuMat/src/DenseLinAlgOps.h:384
MethodDeterminantOp
extensions/include/cuMat/src/DenseLinAlgOps.h:580
MethodDiagonalPreconditioner
extensions/include/cuMat/src/ConjugateGradient.h:227
FunctionELLPACKMVKernel_StaticBatches
extensions/include/cuMat/src/SparseProductEvaluation.h:197
MethodEvent
extensions/include/cuMat/src/Context.h:467
MethodExtractComplexPartOp
extensions/include/cuMat/src/UnaryOps.h:532
MethodExtractComplexPartOp
extensions/include/cuMat/src/UnaryOps.h:590
FunctionExtractComplexPartOp<_Child, false, false> real
extensions/include/cuMat/src/UnaryOps.h:844
FunctionExtractComplexPartOp<_Child, true, false> imag
extensions/include/cuMat/src/UnaryOps.h:851
MethodExtractDiagonalOp
extensions/include/cuMat/src/UnaryOps.h:473
FunctionFDT
(strand)
utils/general_util.py:81
MethodIOFormat
Default constructor, see class IOFormat for the meaning of the parameters */
extensions/include/cuMat/src/IO.h:45
MethodIdentityPreconditioner
extensions/include/cuMat/src/ConjugateGradient.h:257
FunctionInverseKernel
extensions/include/cuMat/src/DenseLinAlgOps.h:488
FunctionInverseKernelWithDet
extensions/include/cuMat/src/DenseLinAlgOps.h:502
MethodInverseOp
extensions/include/cuMat/src/DenseLinAlgOps.h:623
MethodLUDecomposition
* \brief Uninitialized LU decomposition, * but with the specified matrix dimensions. * Since LU decompositions also allow non-symmetric matrices,
extensions/include/cuMat/src/LUDecomposition.h:70
FunctionMatrix
* \brief Default constructor. * For completely fixed-size matrices, this creates a matrix of that size. * For (fully or partially) dynamic matrice
extensions/include/cuMat/src/Matrix.h:503
MethodMatrixBlock
extensions/include/cuMat/src/MatrixBlock.h:50
MethodMatrixInplaceAssignment
extensions/include/cuMat/src/Matrix.h:1352
MethodMatrixReadWrapper
extensions/include/cuMat/src/MatrixBase.h:140
MethodNullaryOp
extensions/include/cuMat/src/NullaryOps.h:54
FunctionPYBIND11_MODULE
extensions/push_pull_inpaint.cpp:50
MethodProductOp
extensions/include/cuMat/src/ProductOp.h:175
FunctionRandomEvaluationKernel
extensions/include/cuMat/src/SimpleRandom.h:112
FunctionReduceBlockKernel
extensions/include/cuMat/src/ReductionOps.h:404
FunctionReduceThreadKernel
extensions/include/cuMat/src/ReductionOps.h:317
FunctionReduceWarpKernel
extensions/include/cuMat/src/ReductionOps.h:356
MethodReductionOp_DynamicSwitched
extensions/include/cuMat/src/ReductionOps.h:708
MethodReductionOp_StaticSwitched
extensions/include/cuMat/src/ReductionOps.h:805
MethodSimpleRandom
* \brief Creates a new random-number generator using the specified seed * \param seed the seed */
extensions/include/cuMat/src/SimpleRandom.h:192
MethodSolveOp
extensions/include/cuMat/src/SolverBase.h:87
MethodSolveWithGuessOp
extensions/include/cuMat/src/IterativeSolverBase.h:185
FunctionSolveWithGuessOp<_SolverImpl, Rhs, Guess> solveWithGuess
extensions/include/cuMat/src/IterativeSolverBase.h:119
MethodSparseExpressionOp
extensions/include/cuMat/src/SparseExpressionOp.h:55
MethodSparseMatrixBase
extensions/include/cuMat/src/SparseMatrixBase.h:388
MethodSparseMatrixDirectAccess
extensions/include/cuMat/src/SparseMatrix.h:568
MethodSparseMatrixInplaceAssignment
extensions/include/cuMat/src/SparseMatrix.h:519
MethodStridedMatrixOutputIterator
extensions/include/cuMat/src/Iterator.h:196
MethodTransposeOp
extensions/include/cuMat/src/TransposeOp.h:132
MethodUnaryOp
extensions/include/cuMat/src/UnaryOps.h:57
Method__call__
(self, tensor)
utils/general_util.py:532
Method__call__
(self, img)
utils/general_util.py:548
Method__call__
(self, t0, t1)
k_diffusion/sampling.py:86
Method__call__
(self, sigma, sigma_next)
k_diffusion/sampling.py:112
Method__call__
(self, *args, **kwargs)
k_diffusion/models/flags.py:55
Method__getitem__
(self, key)
k_diffusion/utils.py:407
Method__getitem__
(self, idx)
data_loader/dataloader.py:397
Method__getitem__
(self, item)
callbacks/callback.py:58
Method__init__
(self)
train_rgb2material.py:35
Method__init__
(self)
train_strandsVAE.py:33
Method__init__
(self, in_channels, out_channels, bias=True)
modules/networks.py:9
← previousnext →301–400 of 981, ranked by callers