MCPcopy Create free account

hub / github.com/csguoh/OBR / functions

Functions645 in github.com/csguoh/OBR

⨍ Functions 645◇ Types & classes 88↳ Endpoints 2
↓ 1 callersFunctionrotate_mlp_output
(layer, R1)
SpinQuant/eval_utils/rotation_utils.py:92
↓ 1 callersFunctionrotate_mlp_output
(layer, Q, model_type)
QuaRot/rotation_utils.py:169
↓ 1 callersFunctionrotate_ov_proj
(layer, head_num, head_dim, R2=None)
SpinQuant/eval_utils/rotation_utils.py:114
↓ 1 callersFunctionrotate_ov_proj
(layer, model_type, head_num, head_dim)
QuaRot/rotation_utils.py:213
↓ 1 callersFunctionset_require_grad_all
(model, requires_grad)
FlatQuant/flatquant/function_utils.py:80
↓ 1 callersFunctionset_seed
(seed)
SpinQuant/utils/utils.py:78
↓ 1 callersFunctionsym_dequant
(q, scale)
SpinQuant/utils/quant_utils.py:53
↓ 1 callersFunctionsym_dequant
(q, scale)
QuaRot/quant_utils.py:34
↓ 1 callersFunctionsym_dequant
(q, scale)
FlatQuant/flatquant/quant_utils.py:25
↓ 1 callersFunctionsym_quant
(x, scale, maxq)
FlatQuant/flatquant/quant_utils.py:19
↓ 1 callersFunctiontest_functionality
Simple functionality test
QuaRot/profiling/compare_with_three_baseline.py:375
↓ 1 callersFunctionthird_party_cmake
()
FlatQuant/setup.py:29
↓ 1 callersMethodtokenize_function
(self, examples)
SpinQuant/utils/data_utils.py:72
↓ 1 callersFunctiontrain
()
SpinQuant/main.py:25
↓ 1 callersFunctiontrain
()
SpinQuant/optimize_rotation.py:42
↓ 1 callersFunctiontwo_compl
(x, bits: int)
QuaRot/quant_utils.py:41
↓ 1 callersFunctionunit
(v, dim: int = 1, eps: float = 1e-8)
SpinQuant/train_utils/optimizer.py:16
↓ 1 callersFunctionwanda_pruning
(weight, scaler_row, sparsity_ratio,prune_n=0,prune_m=0)
SpinQuant/eval_utils/prune_utils.py:6
↓ 1 callersFunctionwanda_pruning
(weight, scaler_row, sparsity_ratio,prune_n=0,prune_m=0)
QuaRot/prune_utils.py:6
↓ 1 callersFunctionwanda_pruning
(weight, scaler_row, sparsity_ratio,prune_n=0,prune_m=0)
FlatQuant/flatquant/prune_utils.py:6
Method__getitem__
(self, i)
SpinQuant/utils/data_utils.py:66
Method__init__
(self, input_ids)
SpinQuant/ppl_utils.py:17
Method__init__
(self, R_init)
SpinQuant/optimize_rotation.py:31
Method__init__
(self, dataset, tokenizer, block_size: int = 1024)
SpinQuant/utils/data_utils.py:47
Method__init__
(self, module: torch.nn.Linear)
SpinQuant/utils/quant_utils.py:210
Method__init__
(self, shape: int = 1)
SpinQuant/utils/quant_utils.py:303
Method__init__
(self, module)
SpinQuant/utils/eval_utils.py:54
Method__init__
(self, layer)
SpinQuant/eval_utils/obr_utils.py:81
Method__init__
(self, module)
SpinQuant/eval_utils/obr_utils.py:300
Method__init__
(self, func, config, *args, **kwargs)
SpinQuant/eval_utils/rotation_utils.py:151
Method__init__
LlamaRMSNorm is equivalent to T5LayerNorm
SpinQuant/eval_utils/modeling_llama.py:124
Method__init__
( self, dim=None, max_position_embeddings=2048, base=10000, device=Non
SpinQuant/eval_utils/modeling_llama.py:147
Method__init__
(self, *args, **kwargs)
SpinQuant/eval_utils/modeling_llama.py:253
Method__init__
(self, *args, **kwargs)
SpinQuant/eval_utils/modeling_llama.py:265
Method__init__
(self, config: LlamaConfig, layer_idx: Optional[int] = None)
SpinQuant/eval_utils/modeling_llama.py:377
Method__init__
(self, *args, **kwargs)
SpinQuant/eval_utils/modeling_llama.py:556
Method__init__
(self, config: LlamaConfig, layer_idx: int)
SpinQuant/eval_utils/modeling_llama.py:800
Method__init__
(self, config: LlamaConfig)
SpinQuant/eval_utils/modeling_llama.py:1019
Method__init__
(self, config)
SpinQuant/eval_utils/modeling_llama.py:1266
Method__init__
(self, config)
SpinQuant/eval_utils/modeling_llama.py:1512
Method__init__
(self, config)
SpinQuant/eval_utils/modeling_llama.py:1642
Method__init__
(self, module)
SpinQuant/eval_utils/gptq_utils.py:196
Method__init__
( self, params, lr, momentum: int = 0, dampening: int = 0, wei
SpinQuant/train_utils/optimizer.py:86
Method__init__
(self, func, config, *args, **kwargs)
SpinQuant/train_utils/apply_r3_r4.py:48
Method__init__
LlamaRMSNorm is equivalent to T5LayerNorm
SpinQuant/train_utils/modeling_llama_quant.py:126
Method__init__
( self, dim=None, max_position_embeddings=2048, base=10000, device=Non
SpinQuant/train_utils/modeling_llama_quant.py:149
Method__init__
(self, *args, **kwargs)
SpinQuant/train_utils/modeling_llama_quant.py:255
Method__init__
(self, *args, **kwargs)
SpinQuant/train_utils/modeling_llama_quant.py:267
Method__init__
(self, config: LlamaConfig, layer_idx: Optional[int] = None)
SpinQuant/train_utils/modeling_llama_quant.py:372
Method__init__
(self, *args, **kwargs)
SpinQuant/train_utils/modeling_llama_quant.py:553
Method__init__
(self, config: LlamaConfig, layer_idx: int)
SpinQuant/train_utils/modeling_llama_quant.py:799
Method__init__
(self, config: LlamaConfig)
SpinQuant/train_utils/modeling_llama_quant.py:1020
Method__init__
(self, config)
SpinQuant/train_utils/modeling_llama_quant.py:1275
Method__init__
(self, config)
SpinQuant/train_utils/modeling_llama_quant.py:1528
Method__init__
(self, config)
SpinQuant/train_utils/modeling_llama_quant.py:1658
Method__init__
( self, model: Union[PreTrainedModel, nn.Module] = None, args: TrainingArguments = Non
SpinQuant/train_utils/fsdp_trainer.py:149
Method__init__
(self, layer)
QuaRot/obr_utils.py:71
Method__init__
(self, module)
QuaRot/obr_utils.py:288
Method__init__
(self, input_ids)
QuaRot/data_utils.py:43
Method__init__
(self, module:torch.nn.Linear)
QuaRot/quant_utils.py:189
Method__init__
(self, shape=1)
QuaRot/quant_utils.py:263
Method__init__
(self, func, config, *args, **kwargs)
QuaRot/rotation_utils.py:266
Method__init__
(self, module)
QuaRot/eval_utils.py:60
Method__init__
(self, input_ids)
QuaRot/ppl_utils.py:20
Method__init__
(self, mean_dim: int, eps=1e-5)
QuaRot/model_utils.py:265
Method__init__
(self, module)
QuaRot/gptq_utils.py:288
Method__init__
(self, module)
FlatQuant/gptq_utils.py:176
Method__init__
(self, module)
FlatQuant/flatquant/obr_utils.py:296
Method__init__
(self, input_ids)
FlatQuant/flatquant/data_utils.py:8
Method__init__
(self, bits, sym=False, lac=False, groupsize=-1, clip_ratio=None, )
FlatQuant/flatquant/quant_utils.py:53
Method__init__
(self, input_ids)
FlatQuant/flatquant/ppl_utils.py:19
Method__init__
(self, left_size, right_size, add_diag=False, diag_init_para=None)
FlatQuant/flatquant/trans_utils.py:58
Method__init__
(self, size)
FlatQuant/flatquant/trans_utils.py:129
Method__init__
(self, left_size, right_size, add_diag=False, diag_init_para=None)
FlatQuant/flatquant/trans_utils.py:171
Method__init__
(self, module)
FlatQuant/flatquant/train_utils.py:44
Method__init__
(self, args, linear: nn.Linear)
FlatQuant/flatquant/flat_linear.py:9
Method__init__
(self, args, module: LlamaAttention)
FlatQuant/flatquant/model_tools/llama31_utils.py:111
Method__init__
(self, args, module: Qwen2Attention)
FlatQuant/flatquant/model_tools/qwen_utils.py:115
Method__init__
(self, flat_args, linear, is_rowparallel=False, act_quantizer=None)
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:22
Method__init__
(self, flat_args, module: MLP)
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:307
Method__init__
(self, flat_args, module, act_quantizer_w1=None, act_quantizer_w2=None)
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:366
Method__init__
(self, flat_args, module: MoE)
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:406
Method__init__
(self, args, module: LlamaAttention)
FlatQuant/flatquant/model_tools/llama_utils.py:111
Method__iter__
(self)
SpinQuant/utils/data_utils.py:69
Method__len__
(self)
SpinQuant/utils/data_utils.py:63
Method__repr__
(self)
FlatQuant/flatquant/trans_utils.py:48
Method__repr__
(self)
FlatQuant/flatquant/trans_utils.py:118
Method__repr__
(self)
FlatQuant/flatquant/trans_utils.py:161
Method__repr__
(self)
FlatQuant/flatquant/trans_utils.py:215
Method__setstate__
(self, state)
SpinQuant/train_utils/optimizer.py:112
Method_init_weights
(self, module)
SpinQuant/eval_utils/modeling_llama.py:920
Method_init_weights
(self, module)
SpinQuant/train_utils/modeling_llama_quant.py:921
Method_ori_forward
(self, hidden_states)
FlatQuant/flatquant/flat_linear.py:42
Method_ori_forward
(self, hidden_states)
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:65
Method_ori_forward
(self, x: torch.Tensor)
FlatQuant/flatquant/model_tools/deepseekv3_utils.py:373
Functionadd_actquant
( module: ActQuantWrapper, name: str = "", layers=[ torch.nn.Linear, QuantizeLinea
SpinQuant/utils/quant_utils.py:502
Functionadd_actquant
(module, name='', layers=[torch.nn.Linear, ActQuantWrapper,
QuaRot/quant_utils.py:379
Functionadd_qk_rotation_wrapper_after_function_call_in_forward
This function adds a rotation wrapper after the output of a function call in forward. Only calls directly in the forward function are affecte
SpinQuant/eval_utils/rotation_utils.py:209
Functionadd_qk_rotation_wrapper_after_function_call_in_forward
This function adds a rotation wrapper after the output of a function call in forward. Only calls directly in the forward function are affecte
SpinQuant/train_utils/apply_r3_r4.py:106
Functionadd_qk_rotation_wrapper_after_function_call_in_forward
This function adds a rotation wrapper after the output of a function call in forward. Only calls directly in the forward function are affect
QuaRot/rotation_utils.py:308
← previousnext →301–400 of 645, ranked by callers