brainstate 0.1.8__py2.py3-none-any.whl → 0.1.9__py2.py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- brainstate/__init__.py +58 -51
- brainstate/_compatible_import.py +148 -148
- brainstate/_state.py +1605 -1663
- brainstate/_state_test.py +52 -52
- brainstate/_utils.py +47 -47
- brainstate/augment/__init__.py +30 -30
- brainstate/augment/_autograd.py +778 -778
- brainstate/augment/_autograd_test.py +1289 -1289
- brainstate/augment/_eval_shape.py +99 -99
- brainstate/augment/_eval_shape_test.py +38 -38
- brainstate/augment/_mapping.py +1060 -1060
- brainstate/augment/_mapping_test.py +597 -597
- brainstate/augment/_random.py +151 -151
- brainstate/compile/__init__.py +38 -38
- brainstate/compile/_ad_checkpoint.py +204 -204
- brainstate/compile/_ad_checkpoint_test.py +49 -49
- brainstate/compile/_conditions.py +256 -256
- brainstate/compile/_conditions_test.py +220 -220
- brainstate/compile/_error_if.py +92 -92
- brainstate/compile/_error_if_test.py +52 -52
- brainstate/compile/_jit.py +346 -346
- brainstate/compile/_jit_test.py +143 -143
- brainstate/compile/_loop_collect_return.py +536 -536
- brainstate/compile/_loop_collect_return_test.py +58 -58
- brainstate/compile/_loop_no_collection.py +184 -184
- brainstate/compile/_loop_no_collection_test.py +50 -50
- brainstate/compile/_make_jaxpr.py +888 -888
- brainstate/compile/_make_jaxpr_test.py +156 -156
- brainstate/compile/_progress_bar.py +202 -202
- brainstate/compile/_unvmap.py +159 -159
- brainstate/compile/_util.py +147 -147
- brainstate/environ.py +563 -563
- brainstate/environ_test.py +62 -62
- brainstate/functional/__init__.py +27 -26
- brainstate/graph/__init__.py +29 -29
- brainstate/graph/_graph_node.py +244 -244
- brainstate/graph/_graph_node_test.py +73 -73
- brainstate/graph/_graph_operation.py +1738 -1738
- brainstate/graph/_graph_operation_test.py +563 -563
- brainstate/init/__init__.py +26 -26
- brainstate/init/_base.py +52 -52
- brainstate/init/_generic.py +244 -244
- brainstate/init/_random_inits.py +553 -553
- brainstate/init/_random_inits_test.py +149 -149
- brainstate/init/_regular_inits.py +105 -105
- brainstate/init/_regular_inits_test.py +50 -50
- brainstate/mixin.py +365 -363
- brainstate/mixin_test.py +77 -73
- brainstate/nn/__init__.py +135 -131
- brainstate/{functional → nn}/_activations.py +808 -813
- brainstate/{functional → nn}/_activations_test.py +331 -331
- brainstate/nn/_collective_ops.py +514 -514
- brainstate/nn/_collective_ops_test.py +43 -43
- brainstate/nn/_common.py +178 -178
- brainstate/nn/_conv.py +501 -501
- brainstate/nn/_conv_test.py +238 -238
- brainstate/nn/_delay.py +509 -502
- brainstate/nn/_delay_test.py +238 -184
- brainstate/nn/_dropout.py +426 -426
- brainstate/nn/_dropout_test.py +100 -100
- brainstate/nn/_dynamics.py +1343 -1343
- brainstate/nn/_dynamics_test.py +78 -78
- brainstate/nn/_elementwise.py +1119 -1119
- brainstate/nn/_elementwise_test.py +169 -169
- brainstate/nn/_embedding.py +58 -58
- brainstate/nn/_exp_euler.py +92 -92
- brainstate/nn/_exp_euler_test.py +35 -35
- brainstate/nn/_fixedprob.py +239 -239
- brainstate/nn/_fixedprob_test.py +114 -114
- brainstate/nn/_inputs.py +608 -608
- brainstate/nn/_linear.py +424 -424
- brainstate/nn/_linear_mv.py +83 -83
- brainstate/nn/_linear_mv_test.py +120 -120
- brainstate/nn/_linear_test.py +107 -107
- brainstate/nn/_ltp.py +28 -28
- brainstate/nn/_module.py +377 -377
- brainstate/nn/_module_test.py +40 -40
- brainstate/nn/_neuron.py +705 -705
- brainstate/nn/_neuron_test.py +161 -161
- brainstate/nn/_normalizations.py +975 -918
- brainstate/nn/_normalizations_test.py +73 -73
- brainstate/{functional → nn}/_others.py +46 -46
- brainstate/nn/_poolings.py +1177 -1177
- brainstate/nn/_poolings_test.py +217 -217
- brainstate/nn/_projection.py +486 -486
- brainstate/nn/_rate_rnns.py +554 -554
- brainstate/nn/_rate_rnns_test.py +63 -63
- brainstate/nn/_readout.py +209 -209
- brainstate/nn/_readout_test.py +53 -53
- brainstate/nn/_stp.py +236 -236
- brainstate/nn/_synapse.py +505 -505
- brainstate/nn/_synapse_test.py +131 -131
- brainstate/nn/_synaptic_projection.py +423 -423
- brainstate/nn/_synouts.py +162 -162
- brainstate/nn/_synouts_test.py +57 -57
- brainstate/nn/_utils.py +89 -89
- brainstate/nn/metrics.py +388 -388
- brainstate/optim/__init__.py +38 -38
- brainstate/optim/_base.py +64 -64
- brainstate/optim/_lr_scheduler.py +448 -448
- brainstate/optim/_lr_scheduler_test.py +50 -50
- brainstate/optim/_optax_optimizer.py +152 -152
- brainstate/optim/_optax_optimizer_test.py +53 -53
- brainstate/optim/_sgd_optimizer.py +1104 -1104
- brainstate/random/__init__.py +24 -24
- brainstate/random/_rand_funs.py +3616 -3616
- brainstate/random/_rand_funs_test.py +567 -567
- brainstate/random/_rand_seed.py +210 -210
- brainstate/random/_rand_seed_test.py +48 -48
- brainstate/random/_rand_state.py +1409 -1409
- brainstate/random/_random_for_unit.py +52 -52
- brainstate/surrogate.py +1957 -1957
- brainstate/transform.py +23 -23
- brainstate/typing.py +304 -304
- brainstate/util/__init__.py +50 -50
- brainstate/util/caller.py +98 -98
- brainstate/util/error.py +55 -55
- brainstate/util/filter.py +469 -469
- brainstate/util/others.py +540 -540
- brainstate/util/pretty_pytree.py +945 -945
- brainstate/util/pretty_pytree_test.py +159 -159
- brainstate/util/pretty_repr.py +328 -328
- brainstate/util/pretty_table.py +2954 -2954
- brainstate/util/scaling.py +258 -258
- brainstate/util/struct.py +523 -523
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/METADATA +91 -99
- brainstate-0.1.9.dist-info/RECORD +130 -0
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/WHEEL +1 -1
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info/licenses}/LICENSE +202 -202
- brainstate/functional/_normalization.py +0 -81
- brainstate/functional/_spikes.py +0 -204
- brainstate-0.1.8.dist-info/RECORD +0 -132
- {brainstate-0.1.8.dist-info → brainstate-0.1.9.dist-info}/top_level.txt +0 -0
brainstate/mixin_test.py
CHANGED
@@ -1,73 +1,77 @@
|
|
1
|
-
# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
|
2
|
-
#
|
3
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
-
# you may not use this file except in compliance with the License.
|
5
|
-
# You may obtain a copy of the License at
|
6
|
-
#
|
7
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
-
#
|
9
|
-
# Unless required by applicable law or agreed to in writing, software
|
10
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
-
# See the License for the specific language governing permissions and
|
13
|
-
# limitations under the License.
|
14
|
-
# ==============================================================================
|
15
|
-
|
16
|
-
import unittest
|
17
|
-
|
18
|
-
import
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
self.assertTrue(
|
27
|
-
self.assertTrue(
|
28
|
-
self.assertTrue(
|
29
|
-
self.assertTrue(
|
30
|
-
self.assertTrue(
|
31
|
-
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
36
|
-
|
37
|
-
|
38
|
-
|
39
|
-
|
40
|
-
self.assertTrue(
|
41
|
-
self.assertTrue(
|
42
|
-
|
43
|
-
|
44
|
-
|
45
|
-
|
46
|
-
|
47
|
-
|
48
|
-
|
49
|
-
self.assertTrue(a.
|
50
|
-
self.
|
51
|
-
self.
|
52
|
-
|
53
|
-
|
54
|
-
a
|
55
|
-
|
56
|
-
|
57
|
-
|
58
|
-
self.assertTrue(a.
|
59
|
-
|
60
|
-
self.
|
61
|
-
self.
|
62
|
-
|
63
|
-
|
64
|
-
a
|
65
|
-
|
66
|
-
|
67
|
-
|
68
|
-
self.assertTrue(a.
|
69
|
-
|
70
|
-
self.
|
71
|
-
self.
|
72
|
-
|
73
|
-
self.assertFalse(a.
|
1
|
+
# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
# ==============================================================================
|
15
|
+
|
16
|
+
import unittest
|
17
|
+
import jax
|
18
|
+
import jax.numpy as jnp
|
19
|
+
import numpy as np
|
20
|
+
|
21
|
+
import brainstate
|
22
|
+
|
23
|
+
|
24
|
+
class TestMixin(unittest.TestCase):
|
25
|
+
def test_mixin(self):
|
26
|
+
self.assertTrue(brainstate.mixin.Mixin)
|
27
|
+
self.assertTrue(brainstate.mixin.ParamDesc)
|
28
|
+
self.assertTrue(brainstate.mixin.ParamDescriber)
|
29
|
+
self.assertTrue(brainstate.mixin.JointTypes)
|
30
|
+
self.assertTrue(brainstate.mixin.OneOfTypes)
|
31
|
+
self.assertTrue(brainstate.mixin.Mode)
|
32
|
+
self.assertTrue(brainstate.mixin.Batching)
|
33
|
+
self.assertTrue(brainstate.mixin.Training)
|
34
|
+
|
35
|
+
|
36
|
+
class TestMode(unittest.TestCase):
|
37
|
+
def test_JointMode(self):
|
38
|
+
a = brainstate.mixin.JointMode(brainstate.mixin.Batching(), brainstate.mixin.Training())
|
39
|
+
self.assertTrue(a.is_a(brainstate.mixin.JointTypes[brainstate.mixin.Batching, brainstate.mixin.Training]))
|
40
|
+
self.assertTrue(a.has(brainstate.mixin.Batching))
|
41
|
+
self.assertTrue(a.has(brainstate.mixin.Training))
|
42
|
+
b = brainstate.mixin.JointMode(brainstate.mixin.Batching())
|
43
|
+
self.assertTrue(b.is_a(brainstate.mixin.JointTypes[brainstate.mixin.Batching]))
|
44
|
+
self.assertTrue(b.is_a(brainstate.mixin.Batching))
|
45
|
+
self.assertTrue(b.has(brainstate.mixin.Batching))
|
46
|
+
|
47
|
+
def test_Training(self):
|
48
|
+
a = brainstate.mixin.Training()
|
49
|
+
self.assertTrue(a.is_a(brainstate.mixin.Training))
|
50
|
+
self.assertTrue(a.is_a(brainstate.mixin.JointTypes[brainstate.mixin.Training]))
|
51
|
+
self.assertTrue(a.has(brainstate.mixin.Training))
|
52
|
+
self.assertTrue(a.has(brainstate.mixin.JointTypes[brainstate.mixin.Training]))
|
53
|
+
self.assertFalse(a.is_a(brainstate.mixin.Batching))
|
54
|
+
self.assertFalse(a.has(brainstate.mixin.Batching))
|
55
|
+
|
56
|
+
def test_Batching(self):
|
57
|
+
a = brainstate.mixin.Batching()
|
58
|
+
self.assertTrue(a.is_a(brainstate.mixin.Batching))
|
59
|
+
self.assertTrue(a.is_a(brainstate.mixin.JointTypes[brainstate.mixin.Batching]))
|
60
|
+
self.assertTrue(a.has(brainstate.mixin.Batching))
|
61
|
+
self.assertTrue(a.has(brainstate.mixin.JointTypes[brainstate.mixin.Batching]))
|
62
|
+
|
63
|
+
self.assertFalse(a.is_a(brainstate.mixin.Training))
|
64
|
+
self.assertFalse(a.has(brainstate.mixin.Training))
|
65
|
+
|
66
|
+
def test_Mode(self):
|
67
|
+
a = brainstate.mixin.Mode()
|
68
|
+
self.assertTrue(a.is_a(brainstate.mixin.Mode))
|
69
|
+
self.assertTrue(a.is_a(brainstate.mixin.JointTypes[brainstate.mixin.Mode]))
|
70
|
+
self.assertTrue(a.has(brainstate.mixin.Mode))
|
71
|
+
self.assertTrue(a.has(brainstate.mixin.JointTypes[brainstate.mixin.Mode]))
|
72
|
+
|
73
|
+
self.assertFalse(a.is_a(brainstate.mixin.Training))
|
74
|
+
self.assertFalse(a.has(brainstate.mixin.Training))
|
75
|
+
self.assertFalse(a.is_a(brainstate.mixin.Batching))
|
76
|
+
self.assertFalse(a.has(brainstate.mixin.Batching))
|
77
|
+
|
brainstate/nn/__init__.py
CHANGED
@@ -1,131 +1,135 @@
|
|
1
|
-
# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
|
2
|
-
#
|
3
|
-
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
-
# you may not use this file except in compliance with the License.
|
5
|
-
# You may obtain a copy of the License at
|
6
|
-
#
|
7
|
-
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
-
#
|
9
|
-
# Unless required by applicable law or agreed to in writing, software
|
10
|
-
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
-
# See the License for the specific language governing permissions and
|
13
|
-
# limitations under the License.
|
14
|
-
# ==============================================================================
|
15
|
-
|
16
|
-
|
17
|
-
from . import metrics
|
18
|
-
from ._collective_ops import *
|
19
|
-
from ._collective_ops import __all__ as collective_ops_all
|
20
|
-
from ._common import *
|
21
|
-
from ._common import __all__ as common_all
|
22
|
-
from ._conv import *
|
23
|
-
from ._conv import __all__ as conv_all
|
24
|
-
from ._delay import *
|
25
|
-
from ._delay import __all__ as state_delay_all
|
26
|
-
from ._dropout import *
|
27
|
-
from ._dropout import __all__ as dropout_all
|
28
|
-
from ._dynamics import *
|
29
|
-
from ._dynamics import __all__ as dyn_all
|
30
|
-
from ._elementwise import *
|
31
|
-
from ._elementwise import __all__ as elementwise_all
|
32
|
-
from ._embedding import *
|
33
|
-
from ._embedding import __all__ as embed_all
|
34
|
-
from ._exp_euler import *
|
35
|
-
from ._exp_euler import __all__ as exp_euler_all
|
36
|
-
from ._fixedprob import *
|
37
|
-
from._fixedprob import __all__ as fixedprob_all
|
38
|
-
from ._inputs import *
|
39
|
-
from ._inputs import __all__ as inputs_all
|
40
|
-
from ._linear import *
|
41
|
-
from ._linear import __all__ as linear_all
|
42
|
-
from ._linear_mv import *
|
43
|
-
from ._linear_mv import __all__ as linear_mv_all
|
44
|
-
from ._ltp import *
|
45
|
-
from ._ltp import __all__ as ltp_all
|
46
|
-
from ._module import *
|
47
|
-
from ._module import __all__ as module_all
|
48
|
-
from ._neuron import *
|
49
|
-
from ._neuron import __all__ as dyn_neuron_all
|
50
|
-
from ._normalizations import *
|
51
|
-
from ._normalizations import __all__ as normalizations_all
|
52
|
-
from .
|
53
|
-
from .
|
54
|
-
from .
|
55
|
-
from .
|
56
|
-
from .
|
57
|
-
from .
|
58
|
-
from .
|
59
|
-
from .
|
60
|
-
from .
|
61
|
-
from .
|
62
|
-
from .
|
63
|
-
from .
|
64
|
-
from .
|
65
|
-
from .
|
66
|
-
from .
|
67
|
-
from .
|
68
|
-
from .
|
69
|
-
from .
|
70
|
-
|
71
|
-
__all__
|
72
|
-
|
73
|
-
|
74
|
-
|
75
|
-
|
76
|
-
|
77
|
-
+
|
78
|
-
+
|
79
|
-
+
|
80
|
-
+
|
81
|
-
+
|
82
|
-
+
|
83
|
-
+
|
84
|
-
+
|
85
|
-
+
|
86
|
-
+
|
87
|
-
+
|
88
|
-
+
|
89
|
-
+
|
90
|
-
+
|
91
|
-
+
|
92
|
-
+
|
93
|
-
+
|
94
|
-
+
|
95
|
-
+
|
96
|
-
+
|
97
|
-
+
|
98
|
-
+
|
99
|
-
+
|
100
|
-
+
|
101
|
-
+
|
102
|
-
|
103
|
-
|
104
|
-
|
105
|
-
|
106
|
-
|
107
|
-
|
108
|
-
|
109
|
-
|
110
|
-
|
111
|
-
|
112
|
-
|
113
|
-
|
114
|
-
|
115
|
-
|
116
|
-
|
117
|
-
|
118
|
-
|
119
|
-
|
120
|
-
|
121
|
-
|
122
|
-
|
123
|
-
|
124
|
-
|
125
|
-
|
126
|
-
|
127
|
-
|
128
|
-
|
129
|
-
|
130
|
-
|
131
|
-
|
1
|
+
# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
# ==============================================================================
|
15
|
+
|
16
|
+
|
17
|
+
from . import metrics
|
18
|
+
from ._collective_ops import *
|
19
|
+
from ._collective_ops import __all__ as collective_ops_all
|
20
|
+
from ._common import *
|
21
|
+
from ._common import __all__ as common_all
|
22
|
+
from ._conv import *
|
23
|
+
from ._conv import __all__ as conv_all
|
24
|
+
from ._delay import *
|
25
|
+
from ._delay import __all__ as state_delay_all
|
26
|
+
from ._dropout import *
|
27
|
+
from ._dropout import __all__ as dropout_all
|
28
|
+
from ._dynamics import *
|
29
|
+
from ._dynamics import __all__ as dyn_all
|
30
|
+
from ._elementwise import *
|
31
|
+
from ._elementwise import __all__ as elementwise_all
|
32
|
+
from ._embedding import *
|
33
|
+
from ._embedding import __all__ as embed_all
|
34
|
+
from ._exp_euler import *
|
35
|
+
from ._exp_euler import __all__ as exp_euler_all
|
36
|
+
from ._fixedprob import *
|
37
|
+
from ._fixedprob import __all__ as fixedprob_all
|
38
|
+
from ._inputs import *
|
39
|
+
from ._inputs import __all__ as inputs_all
|
40
|
+
from ._linear import *
|
41
|
+
from ._linear import __all__ as linear_all
|
42
|
+
from ._linear_mv import *
|
43
|
+
from ._linear_mv import __all__ as linear_mv_all
|
44
|
+
from ._ltp import *
|
45
|
+
from ._ltp import __all__ as ltp_all
|
46
|
+
from ._module import *
|
47
|
+
from ._module import __all__ as module_all
|
48
|
+
from ._neuron import *
|
49
|
+
from ._neuron import __all__ as dyn_neuron_all
|
50
|
+
from ._normalizations import *
|
51
|
+
from ._normalizations import __all__ as normalizations_all
|
52
|
+
from ._others import *
|
53
|
+
from ._others import __all__ as _others_all
|
54
|
+
from ._poolings import *
|
55
|
+
from ._poolings import __all__ as poolings_all
|
56
|
+
from ._projection import *
|
57
|
+
from ._projection import __all__ as projection_all
|
58
|
+
from ._rate_rnns import *
|
59
|
+
from ._rate_rnns import __all__ as rate_rnns
|
60
|
+
from ._readout import *
|
61
|
+
from ._readout import __all__ as readout_all
|
62
|
+
from ._stp import *
|
63
|
+
from ._stp import __all__ as stp_all
|
64
|
+
from ._synapse import *
|
65
|
+
from ._synapse import __all__ as dyn_synapse_all
|
66
|
+
from ._synaptic_projection import *
|
67
|
+
from ._synaptic_projection import __all__ as _syn_proj_all
|
68
|
+
from ._synouts import *
|
69
|
+
from ._synouts import __all__ as synouts_all
|
70
|
+
from ._utils import *
|
71
|
+
from ._utils import __all__ as utils_all
|
72
|
+
|
73
|
+
__all__ = (
|
74
|
+
[
|
75
|
+
'metrics',
|
76
|
+
]
|
77
|
+
+ collective_ops_all
|
78
|
+
+ common_all
|
79
|
+
+ elementwise_all
|
80
|
+
+ module_all
|
81
|
+
+ exp_euler_all
|
82
|
+
+ utils_all
|
83
|
+
+ dyn_all
|
84
|
+
+ projection_all
|
85
|
+
+ state_delay_all
|
86
|
+
+ synouts_all
|
87
|
+
+ conv_all
|
88
|
+
+ linear_all
|
89
|
+
+ normalizations_all
|
90
|
+
+ poolings_all
|
91
|
+
+ fixedprob_all
|
92
|
+
+ linear_mv_all
|
93
|
+
+ embed_all
|
94
|
+
+ dropout_all
|
95
|
+
+ elementwise_all
|
96
|
+
+ dyn_neuron_all
|
97
|
+
+ dyn_synapse_all
|
98
|
+
+ inputs_all
|
99
|
+
+ rate_rnns
|
100
|
+
+ readout_all
|
101
|
+
+ stp_all
|
102
|
+
+ ltp_all
|
103
|
+
+ _syn_proj_all
|
104
|
+
+ _others_all
|
105
|
+
)
|
106
|
+
|
107
|
+
del (
|
108
|
+
collective_ops_all,
|
109
|
+
common_all,
|
110
|
+
module_all,
|
111
|
+
exp_euler_all,
|
112
|
+
utils_all,
|
113
|
+
dyn_all,
|
114
|
+
projection_all,
|
115
|
+
state_delay_all,
|
116
|
+
synouts_all,
|
117
|
+
conv_all,
|
118
|
+
linear_all,
|
119
|
+
normalizations_all,
|
120
|
+
poolings_all,
|
121
|
+
embed_all,
|
122
|
+
fixedprob_all,
|
123
|
+
linear_mv_all,
|
124
|
+
dropout_all,
|
125
|
+
elementwise_all,
|
126
|
+
dyn_neuron_all,
|
127
|
+
dyn_synapse_all,
|
128
|
+
inputs_all,
|
129
|
+
readout_all,
|
130
|
+
rate_rnns,
|
131
|
+
stp_all,
|
132
|
+
ltp_all,
|
133
|
+
_syn_proj_all,
|
134
|
+
_others_all,
|
135
|
+
)
|