MCPcopy Create free account

hub / github.com/Meshcapade/difflocks / functions

Functions981 in github.com/Meshcapade/difflocks

Methodforward
(self, x)
k_diffusion/models/modules.py:604
Methodforward
(self, x)
k_diffusion/models/modules.py:617
Methodforward
(self, x, skip)
k_diffusion/models/modules.py:629
Methodforward
(self, x, noise_cond)
k_diffusion/models/modules.py:649
Methodforward
(self, x, pos, global_cond, context=None, context_pos=None)
k_diffusion/models/attention.py:109
Methodforward
(self, strands_tbn, strands_positions, root_normals)
data_loader/mesh_utils.py:407
Methodforward
(self, gt_dict)
models/strand_codec.py:156
Methodforward
(self, strand_features, hyperparams, normalization_dict)
models/strand_codec.py:296
Methodforward
(self, gt_dict, hyperparams, normalization_dict)
models/strand_codec.py:415
Methodforward
(self, batch_dict)
models/rgb_to_material.py:51
Methodforward
(self, phase, gt_dict, pred_dict, latent_dict, hyperparams)
losses/loss.py:14
MethodfreeHost
* \brief Frees memory previously allocated with allocateHost(size_t). * Passing a NULL-pointer should be a no-op. * \param memory the memory to be
extensions/include/cuMat/src/Context.h:293
Functionfreqs_pixel
(max_freq=10.0)
k_diffusion/models/axial_rope.py:71
FunctionfromArray
extensions/include/cuMat/src/Matrix.h:809
FunctionfromEigen
* \brief Converts the specified Eigen matrix into the * corresponding cuMat matrix. * Note that Eigen does not support batched matrices. Hence, this
extensions/include/cuMat/src/Matrix.h:922
Functionfrom_pil_image
Converts from a PIL image to a tensor.
k_diffusion/utils.py:20
Functionfw_cat
(x, fw)
utils/resize_right/resize_right.py:372
Functionfw_floor
(x, fw)
utils/resize_right/resize_right.py:365
MethodgetAliveDevicePointers
For testing only
extensions/include/cuMat/src/Context.h:333
MethodgetAliveHostPointers
For testing only
extensions/include/cuMat/src/Context.h:331
FunctiongetCounter
* \brief Returns the current value of the reference counter. * This can be used to determine if this memory is used uniquely * by an object. * \
extensions/include/cuMat/src/DevicePointer.h:147
MethodgetErrorName
extensions/include/cuMat/src/CusolverApi.h:35
MethodgetErrorName
extensions/include/cuMat/src/CublasApi.h:35
MethodgetFreeDeviceMemory
* \brief Returns the free memory on the device in bytes. * \return the free memory in bytes */
extensions/include/cuMat/src/Context.h:340
MethodgetLeft
extensions/include/cuMat/src/BinaryOps.h:146
FunctiongetNNZ
* \return the number of non-zero entries */
extensions/include/cuMat/src/SparseMatrixBase.h:64
MethodgetNNZ
extensions/include/cuMat/src/SparseMatrixBase.h:184
MethodgetRight
extensions/include/cuMat/src/BinaryOps.h:154
MethodgetTotalDeviceMemory
* \brief Returns the total available memory on the device in bytes. * \return the total available memory in bytes */
extensions/include/cuMat/src/Context.h:351
Methodget_base_mesh
(self)
data_loader/dataloader.py:649
Functionget_checkpointing
()
k_diffusion/models/flags.py:30
Methodget_lr
(self)
k_diffusion/utils.py:183
Methodget_lr
(self)
k_diffusion/utils.py:224
Methodget_lr
(self)
k_diffusion/utils.py:256
Methodget_lr
(self)
schedulers/warmup.py:26
Methodget_lr
(self)
schedulers/linearlr.py:63
Methodget_lr
(self)
schedulers/multisteplr.py:50
Functionget_safetensors_metadata
Retrieves the metadata from a safetensors file.
k_diffusion/utils.py:447
Functionget_sigmas_exponential
Constructs an exponential noise schedule.
k_diffusion/sampling.py:26
Functionget_sigmas_karras
Constructs the noise schedule of Karras et al. (2022).
k_diffusion/sampling.py:17
Functionget_sigmas_polyexponential
Constructs an polynomial in log sigma noise schedule.
k_diffusion/sampling.py:32
Functionget_sigmas_vp
Constructs a continuous VP noise schedule.
k_diffusion/sampling.py:39
Functionget_transform
(size=(512,512),apply_agmentationt=False,normalization=False,flip=False)
utils/general_util.py:551
Functionget_use_flash_attention_2
()
k_diffusion/models/flags.py:13
Functionhf_datasets_augs_helper
Apply passed in transforms for HuggingFace Datasets.
k_diffusion/utils.py:38
Functionimag
* \brief Extracts the imaginary part of the complex matrix. * This method is only available for complex matrices. * This is the non-const lval
extensions/include/cuMat/src/Matrix.h:1286
Functionindex
* \brief Converts from the linear index back to row, column and batch index. * Requirement of \c AccessFlags::WriteCwise * \param index the linear
extensions/include/cuMat/src/SparseMatrix.h:152
Methodindex
extensions/include/cuMat/src/UnaryOps.h:622
Methodinner
extensions/include/cuMat/src/ReductionAlgorithmSelection.h:70
Functioninplace
* \brief Forces inplace assignment. * The only usecase is <code>matrix.inplace() = expression</code> * where the content's of matrix are overw
extensions/include/cuMat/src/SparseMatrix.h:398
Functioninverse
* \brief Computes the inverse of the input matrix. * \return The inverse matrix */
extensions/include/cuMat/src/DecompositionBase.h:25
MethodisSingular
* \brief Tests if the matrix at the specified batch was singular during the initial LU decomposition * \param batch the batch * \return true
extensions/include/cuMat/src/LUDecomposition.h:145
MethoditerIn
extensions/include/cuMat/src/ReductionOps.h:64
MethoditerIn
extensions/include/cuMat/src/ReductionOps.h:79
MethoditerIn
extensions/include/cuMat/src/ReductionOps.h:94
MethoditerIn
extensions/include/cuMat/src/ReductionOps.h:115
MethoditerIn
extensions/include/cuMat/src/ReductionOps.h:133
MethoditerIn
extensions/include/cuMat/src/ReductionOps.h:149
MethoditerOut
extensions/include/cuMat/src/ReductionOps.h:68
MethoditerOut
extensions/include/cuMat/src/ReductionOps.h:83
MethoditerOut
extensions/include/cuMat/src/ReductionOps.h:101
MethoditerOut
extensions/include/cuMat/src/ReductionOps.h:122
MethoditerOut
extensions/include/cuMat/src/ReductionOps.h:138
MethoditerOut
extensions/include/cuMat/src/ReductionOps.h:154
Functioniterations
\returns the number of iterations performed during the last solve */
extensions/include/cuMat/src/IterativeSolverBase.h:98
Functionlanczos2
(x)
utils/resize_right/interp_methods.py:47
Functionlanczos3
(x)
utils/resize_right/interp_methods.py:54
Functionlinear
(x)
utils/resize_right/interp_methods.py:61
MethodlinearToCoords
* \brief Converts linear index of nnz to row+col+batch index. This is required for AccessFlags::WriteCwise */
extensions/include/cuMat/src/SparseMatrixBase.h:236
MethodlinearToCoords
* \brief Converts linear index of nnz to row+col+batch index. This is required for AccessFlags::WriteCwise */
extensions/include/cuMat/src/SparseMatrixBase.h:282
MethodlinearToCoords
* \brief Converts linear index of nnz to row+col+batch index. This is required for AccessFlags::WriteCwise */
extensions/include/cuMat/src/SparseMatrixBase.h:325
Methodload
extensions/include/cuMat/src/DenseLinAlgOps.h:55
Methodload
extensions/include/cuMat/src/DenseLinAlgOps.h:80
Methodload
extensions/include/cuMat/src/DenseLinAlgOps.h:111
Methodload
extensions/include/cuMat/src/DenseLinAlgOps.h:151
Methodload
extensions/include/cuMat/src/DenseLinAlgOps.h:199
Methodload
extensions/include/cuMat/src/DenseLinAlgOps.h:260
Functionload_config
(path_or_dict)
k_diffusion/config.py:23
Methodload_state_dict
Loads the warmup scheduler's state. Arguments: state_dict (dict): warmup scheduler state. Should be an object returned
schedulers/pytorch_warmup/base.py:38
Functionload_strand
(file, return_strands=False,interp=False,num_interp=100)
utils/strand_util.py:10
Functionload_strands_from_ply
(file,move_strands_to_usc=True)
utils/strand_util.py:121
FunctionlogDeterminant
* \brief Computes the log-determinant of this matrix. * \return The log-determinant */
extensions/include/cuMat/src/DecompositionBase.h:49
MethodlogDeterminant
* \brief Computes the log-determinant of this matrix. * \return The log-determinant */
extensions/include/cuMat/src/CholeskyDecomposition.h:144
Functionlog_likelihood
(model, x, sigma_min, sigma_max, extra_args=None, atol=1e-4, rtol=1e-4)
k_diffusion/sampling.py:281
Methodloss
(self, input, noise, sigma, **kwargs)
k_diffusion/layers.py:230
Methodloss
(self, input, noise, sigma, **kwargs)
k_diffusion/layers.py:244
FunctionmakeExclusiveUse
* \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/Matrix.h:759
Functionmake_denoiser_wrapper
(config)
k_diffusion/config.py:385
Functionmake_host_scalar
extensions/include/cuMat/src/NullaryOps.h:105
Functionmake_model
(config)
k_diffusion/config.py:213
Functionmake_sample_density
(config)
k_diffusion/config.py:405
MethodmallocHost
* \brief Allocates size-number of bytes on the host system. * This memory must be freed with freeHost(void*). * * Note that allocate zero bytes
extensions/include/cuMat/src/Context.h:245
Functionmap_range_val
( input_val, input_start, input_end, output_start, output_end)
utils/general_util.py:13
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:137
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:142
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:147
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:152
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:157
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:162
Methodmax
extensions/include/cuMat/src/SimpleRandom.h:167
← previousnext →701–800 of 981, ranked by callers