pre2post_sum(pre_values, post_num, post_ids)
|
The pre-to-post synaptic summation. |
pre2post_prod(pre_values, post_num, post_ids)
|
The pre-to-post synaptic production. |
pre2post_max(pre_values, post_num, post_ids)
|
The pre-to-post synaptic maximization. |
pre2post_min(pre_values, post_num, post_ids)
|
The pre-to-post synaptic minimization. |
pre2post_mean(pre_values, post_num, post_ids)
|
The pre-to-post synaptic mean computation. |
pre2syn(pre_values, pre_ids)
|
The pre-to-syn computation. |
syn2post_sum(syn_values, post_ids, post_num)
|
The syn-to-post summation computation. |
syn2post(syn_values, post_ids, post_num[, ...])
|
The syn-to-post summation computation. |
syn2post_prod(syn_values, post_ids, post_num)
|
The syn-to-post product computation. |
syn2post_max(syn_values, post_ids, post_num)
|
The syn-to-post maximum computation. |
syn2post_min(syn_values, post_ids, post_num)
|
The syn-to-post minimization computation. |
syn2post_mean(syn_values, post_ids, post_num)
|
The syn-to-post mean computation. |
syn2post_softmax(syn_values, post_ids, post_num)
|
The syn-to-post softmax computation. |
pre2post_event_sum(events, pre2post, post_num)
|
The pre-to-post synaptic computation with event-driven summation. |
pre2post_event_prod(events, pre2post, post_num)
|
The pre-to-post synaptic computation with event-driven production. |
sparse_matmul(A, B)
|
Sparse matrix multiplication. |
segment_sum(data, segment_ids[, ...])
|
- rtype
JaxArray
|
segment_prod(data, segment_ids[, ...])
|
- rtype
JaxArray
|
segment_max(data, segment_ids[, ...])
|
- rtype
JaxArray
|
segment_min(data, segment_ids[, ...])
|
- rtype
JaxArray
|
register_op(op_name, cpu_func[, gpu_func, ...])
|
Converting the numba-jitted function in a Jax/XLA compatible primitive. |
