Functions — github.com/BICLab/SpikingBrain-7B

hub / github.com/BICLab/SpikingBrain-7B / functions

Functions395 in github.com/BICLab/SpikingBrain-7B

⨍ Functions 395◇ Types & classes 101↳ Endpoints 1

↓ 1 callersMethodneuronal_fire

(self,)

W8ASpike/Int2Spike/neuron.py:140

↓ 1 callersMethodneuronal_fire

(self,)

W8ASpike/Int2Spike/neuron.py:187

↓ 1 callersMethodneuronal_fire

(self)

W8ASpike/Int2Spike/neuron.py:271

↓ 1 callersMethodneuronal_reset

(self, spike)

W8ASpike/Int2Spike/neuron.py:144

↓ 1 callersMethodneuronal_reset

(self, spike)

W8ASpike/Int2Spike/neuron.py:190

↓ 1 callersMethodneuronal_reset

(self, spike)

W8ASpike/Int2Spike/neuron.py:275

↓ 1 callersFunctionpatch_fla_bitnet

()

docker_build/patch_fla_bitnet.py:12

↓ 1 callersFunctionquant

Convert float tensor to spike sequence via integer quantizer + LIF encoder. Args: x (torch.Tensor): Float input tensor. quan

W8ASpike/Int2Spike/neuron.py:380

↓ 1 callersFunctionread_readme

Read the README file if present.

setup.py:19

↓ 1 callersFunctionrms_norm_fn

( x, weight, bias, residual=None, eps=1e-6, prenorm=False, residual_in_fp32=False

hf_7B_model/layernorm.py:515

↓ 1 callersMethodrot_pos_emb

(self, grid_thw)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:413

↓ 1 callersMethodset

(self, layer_idx: int, state: Tuple[torch.Tensor, ...])

hf_7B_VLM/SpikingBrain_VL/utils.py:11

↓ 1 callersFunctionspike_dequant

Decode a spike sequence back to integer values based on encoding type. Args: spike (torch.Tensor): Spike sequence of shape [T, *x.sh

W8ASpike/Int2Spike/neuron.py:326

↓ 1 callersFunctionspike_matmul

Perform matrix multiplication for spike-based inputs with optional spike sequence conversion (LIF nodes) and optional bitwise processing.

W8ASpike/Int2Spike/neuron.py:451

↓ 1 callersFunctionsqrelu_bwd

(g, x)

W8ASpike/activations.py:284

↓ 1 callersFunctionsqrelu_bwd

(g, x)

hf_7B_model/activations.py:284

↓ 1 callersFunctionsqrelu_fwd

(x)

W8ASpike/activations.py:278

↓ 1 callersFunctionsqrelu_fwd

(x)

hf_7B_model/activations.py:278

↓ 1 callersFunctionswiglu_bwd

Computes the backward pass for the SwiGLU activation. Args: x (torch.Tensor): Input tensor x. y (torch.Tensor): Input te

W8ASpike/activations.py:333

↓ 1 callersFunctionswiglu_bwd

Computes the backward pass for the SwiGLU activation. Args: x (torch.Tensor): Input tensor x. y (torch.Tensor): Input te

hf_7B_model/activations.py:333

↓ 1 callersFunctionswiglu_bwd_with_output

Computes the backward pass for the SwiGLU activation, including the intermediate output. Args: x (torch.Tensor): Input tensor x.

W8ASpike/activations.py:352

↓ 1 callersFunctionswiglu_bwd_with_output

Computes the backward pass for the SwiGLU activation, including the intermediate output. Args: x (torch.Tensor): Input tensor x.

hf_7B_model/activations.py:352

Method__call__

( self, images: ImageInput = None, text: Union[TextInput, PreTokenizedInput, List[Text

hf_7B_VLM/SpikingBrain_VL/processing_spikingbrain_vl.py:36

Method__getitem__

(self, layer_idx: int)

W8ASpike/cache.py:27

Method__getitem__

(self, layer_idx: int)

hf_7B_model/cache.py:27

Method__init__

(self, hidden_size, num_heads, num_key_value_heads)

hf_7B_VLM/SpikingBrain_VL/gla.py:29

Method__init__

(self, config, bias: bool = False)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:50

Method__init__

( self, patch_size: int = 14, temporal_patch_size: int = 2, in_channels: int =

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:64

Method__init__

(self, dim: int, theta: float = 10000.0)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:90

Method__init__

SpikingBrainRMSNorm is equivalent to T5LayerNorm

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:102

Method__init__

(self, dim: int, context_dim: int, spatial_merge_size: int = 2)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:122

Method__init__

(self, dim: int, num_heads: int = 16)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:150

Method__init__

(self, dim: int, num_heads: int = 16)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:211

Method__init__

(self, dim: int, num_heads: int = 16)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:261

Method__init__

(self, config, attn_implementation: str = "sdpa")

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:311

Method__init__

(self, config, *inputs, **kwargs)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:385

Method__init__

(self, config: SpikingBrain_VLConfig, device=None)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:544

Method__init__

(self, config: SpikingBrain_VLConfig, layer_idx: Optional[int] = None)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:683

Method__init__

(self, *args, **kwargs)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:793

Method__init__

(self, *args, **kwargs)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:904

Method__init__

(self, config: SpikingBrain_VLConfig, layer_idx: int)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:966

Method__init__

(self, config: SpikingBrain_VLConfig)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:1060

Method__init__

(self, config)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:1474

Method__init__

(self)

hf_7B_VLM/SpikingBrain_VL/utils.py:7

Method__init__

(self, image_processor=None, tokenizer=None, chat_template=None, **kwargs)

hf_7B_VLM/SpikingBrain_VL/processing_spikingbrain_vl.py:31

Method__init__

( self, depth=32, hidden_size=3584, hidden_act="silu", intermediate_si

hf_7B_VLM/SpikingBrain_VL/configuration_spikingbrain_vl.py:10

Method__init__

( self, vocab_size=152064, hidden_size=3584, num_hidden_layers=28, att

W8ASpike/configuration_gla_swa.py:29

Method__init__

(self, config: GLAswaConfig, layer_idx: int)

W8ASpike/modeling_gla_swa.py:59

Method__init__

(self, *inputs, **kwargs)

W8ASpike/modeling_gla_swa.py:135

Method__init__

(self, config: GLAswaConfig)

W8ASpike/modeling_gla_swa.py:174

Method__init__

(self, config)

W8ASpike/modeling_gla_swa.py:301

Method__init__

(self, dim, max_position_embeddings=2048, base=10000, device=None)

W8ASpike/window_attention.py:24

Method__init__

( self, seen_tokens: int = 0 )

W8ASpike/cache.py:19

Method__init__

RMSNorm is equivalent to T5LayerNorm

W8ASpike/norms.py:5

Method__init__

( self, mode: str = 'chunk', hidden_size: int = 1024, num_heads: Optional[int]

W8ASpike/gla_attention.py:37

Method__init__

(self, in_features: int, out_features: int, bias: bool = True, device=None, dtype=None, w_group_size=128, dyna

W8ASpike/quant_linear.py:22

Method__init__

(self,)

W8ASpike/Int2Spike/neuron.py:103

Method__init__

(self,)

W8ASpike/Int2Spike/neuron.py:153

Method__init__

(self, is_bidirectional: bool = False, is_two_complement: bool = False)

W8ASpike/Int2Spike/neuron.py:209

Method__init__

( self, hidden_size: int, elementwise_affine: bool = True, bias: bool = False,

hf_7B_model/layernorm.py:538

Method__init__

( self, num_groups: int, hidden_size: int, elementwise_affine: bool = True,

hf_7B_model/layernorm.py:580

Method__init__

( self, hidden_size, elementwise_affine: bool = True, bias: bool = False,

hf_7B_model/layernorm.py:804

Method__init__

( self, num_groups: int, hidden_size: int, elementwise_affine: bool = True,

hf_7B_model/layernorm.py:849

Method__init__

( self, hidden_size, elementwise_affine: bool = True, bias: bool = False,

hf_7B_model/layernorm.py:900

Method__init__

( self, vocab_size=152064, hidden_size=3584, num_hidden_layers=28, att

hf_7B_model/configuration_gla_swa.py:29

Method__init__

(self, config: GLAswaConfig, layer_idx: int)

hf_7B_model/modeling_gla_swa.py:58

Method__init__

(self, *inputs, **kwargs)

hf_7B_model/modeling_gla_swa.py:135

Method__init__

(self, config: GLAswaConfig)

hf_7B_model/modeling_gla_swa.py:174

Method__init__

(self, config)

hf_7B_model/modeling_gla_swa.py:299

Method__init__

(self, dim, max_position_embeddings=2048, base=10000, device=None)

hf_7B_model/window_attention.py:24

Method__init__

( self, seen_tokens: int = 0 )

hf_7B_model/cache.py:19

Method__init__

RMSNorm is equivalent to T5LayerNorm

hf_7B_model/norms.py:5

Method__init__

( self, mode: str = 'chunk', hidden_size: int = 1024, num_heads: Optional[int]

hf_7B_model/gla_attention.py:37

Method__init__

(self, dim, max_position_embeddings=2048, base=10000, device=None)

hf_7B_model/window_attention_sdpa.py:24

Method__init__

(self, input_builder: "ModelInputForGPUBuilder")

vllm_hymeta/attention/meta_attention.py:415

Method__init__

( self, num_heads: int, head_size: int, scale: float, num_kv_heads: in

vllm_hymeta/attention/meta_attention.py:640

Method__init__

( self, vocab_size=152064, hidden_size=3584, num_hidden_layers=28, att

vllm_hymeta/model_for_7B/configuration_gla_swa.py:15

Method__init__

( self, head_size: int, rotary_dim: int, max_position: int, base: floa

vllm_hymeta/model_for_7B/modeling_gla_swa.py:46

Method__init__

( self, config: GLAswaConfig, hidden_size: int, num_heads: int, head_d

vllm_hymeta/model_for_7B/modeling_gla_swa.py:135

Method__init__

( self, config: GLAswaConfig, layer_idx: int, quant_config: Optional[Quantizat

vllm_hymeta/model_for_7B/modeling_gla_swa.py:223

Method__init__

( self, config: GLAswaConfig, quant_config: Optional[QuantizationConfig] = None,

vllm_hymeta/model_for_7B/modeling_gla_swa.py:311

Method__init__

(self, *, vllm_config: VllmConfig, prefix: str = "")

vllm_hymeta/model_for_7B/modeling_gla_swa.py:509

Method__init__

(self, dtype, cache_shape)

vllm_hymeta/model_for_7B/gla_cache.py:19

Method__init__

( self, input_size: int = 1024, low_rank_dim: int = 16, output_size: int = 102

vllm_hymeta/model_for_7B/gla_attention.py:55

hf_7B_model/cache.py:33

hf_7B_model/cache.py:37

Method__repr__

(self)

hf_7B_model/layernorm.py:558

Method__repr__

(self)

hf_7B_model/layernorm.py:605

Method__repr__

(self)

hf_7B_model/layernorm.py:648

Method__repr__

(self)

hf_7B_model/layernorm.py:824

Method__repr__

(self)

hf_7B_model/layernorm.py:874

Method__repr__

(self)

hf_7B_model/layernorm.py:920

Method_copy_cache

(self, from_index: int, to_index: int)

vllm_hymeta/model_for_7B/gla_cache.py:29

Method_expand_dict_for_generation

(dict_to_expand)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:2020

Method_expand_dict_for_generation_visual

(dict_to_expand)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:1972

Method_expand_inputs_for_generation

( self, expand_size: int = 1, is_encoder_decoder: bool = False, input_ids: Opt

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:1955

Method_init_weights

(self, module)

hf_7B_VLM/SpikingBrain_VL/modeling_spikingbrain_vl.py:369

Method_init_weights

( self, module: nn.Module, rescale_prenorm_residual: bool = True, num_residual

W8ASpike/modeling_gla_swa.py:138

← previous next →101–200 of 395, ranked by callers