brainstate 0.1.8__py2.py3-none-any.whl → 0.1.10__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.
Files changed (133) hide show
  1. brainstate/__init__.py +58 -51
  2. brainstate/_compatible_import.py +148 -148
  3. brainstate/_state.py +1605 -1663
  4. brainstate/_state_test.py +52 -52
  5. brainstate/_utils.py +47 -47
  6. brainstate/augment/__init__.py +30 -30
  7. brainstate/augment/_autograd.py +778 -778
  8. brainstate/augment/_autograd_test.py +1289 -1289
  9. brainstate/augment/_eval_shape.py +99 -99
  10. brainstate/augment/_eval_shape_test.py +38 -38
  11. brainstate/augment/_mapping.py +1060 -1060
  12. brainstate/augment/_mapping_test.py +597 -597
  13. brainstate/augment/_random.py +151 -151
  14. brainstate/compile/__init__.py +38 -38
  15. brainstate/compile/_ad_checkpoint.py +204 -204
  16. brainstate/compile/_ad_checkpoint_test.py +49 -49
  17. brainstate/compile/_conditions.py +256 -256
  18. brainstate/compile/_conditions_test.py +220 -220
  19. brainstate/compile/_error_if.py +92 -92
  20. brainstate/compile/_error_if_test.py +52 -52
  21. brainstate/compile/_jit.py +346 -346
  22. brainstate/compile/_jit_test.py +143 -143
  23. brainstate/compile/_loop_collect_return.py +536 -536
  24. brainstate/compile/_loop_collect_return_test.py +58 -58
  25. brainstate/compile/_loop_no_collection.py +184 -184
  26. brainstate/compile/_loop_no_collection_test.py +50 -50
  27. brainstate/compile/_make_jaxpr.py +888 -888
  28. brainstate/compile/_make_jaxpr_test.py +156 -156
  29. brainstate/compile/_progress_bar.py +202 -202
  30. brainstate/compile/_unvmap.py +159 -159
  31. brainstate/compile/_util.py +147 -147
  32. brainstate/environ.py +563 -563
  33. brainstate/environ_test.py +62 -62
  34. brainstate/functional/__init__.py +27 -26
  35. brainstate/graph/__init__.py +29 -29
  36. brainstate/graph/_graph_node.py +244 -244
  37. brainstate/graph/_graph_node_test.py +73 -73
  38. brainstate/graph/_graph_operation.py +1738 -1738
  39. brainstate/graph/_graph_operation_test.py +563 -563
  40. brainstate/init/__init__.py +26 -26
  41. brainstate/init/_base.py +52 -52
  42. brainstate/init/_generic.py +244 -244
  43. brainstate/init/_random_inits.py +553 -553
  44. brainstate/init/_random_inits_test.py +149 -149
  45. brainstate/init/_regular_inits.py +105 -105
  46. brainstate/init/_regular_inits_test.py +50 -50
  47. brainstate/mixin.py +365 -363
  48. brainstate/mixin_test.py +77 -73
  49. brainstate/nn/__init__.py +135 -131
  50. brainstate/{functional → nn}/_activations.py +808 -813
  51. brainstate/{functional → nn}/_activations_test.py +331 -331
  52. brainstate/nn/_collective_ops.py +514 -514
  53. brainstate/nn/_collective_ops_test.py +43 -43
  54. brainstate/nn/_common.py +178 -178
  55. brainstate/nn/_conv.py +501 -501
  56. brainstate/nn/_conv_test.py +238 -238
  57. brainstate/nn/_delay.py +588 -502
  58. brainstate/nn/_delay_test.py +238 -184
  59. brainstate/nn/_dropout.py +426 -426
  60. brainstate/nn/_dropout_test.py +100 -100
  61. brainstate/nn/_dynamics.py +1343 -1343
  62. brainstate/nn/_dynamics_test.py +78 -78
  63. brainstate/nn/_elementwise.py +1119 -1119
  64. brainstate/nn/_elementwise_test.py +169 -169
  65. brainstate/nn/_embedding.py +58 -58
  66. brainstate/nn/_exp_euler.py +92 -92
  67. brainstate/nn/_exp_euler_test.py +35 -35
  68. brainstate/nn/_fixedprob.py +239 -239
  69. brainstate/nn/_fixedprob_test.py +114 -114
  70. brainstate/nn/_inputs.py +608 -608
  71. brainstate/nn/_linear.py +424 -424
  72. brainstate/nn/_linear_mv.py +83 -83
  73. brainstate/nn/_linear_mv_test.py +120 -120
  74. brainstate/nn/_linear_test.py +107 -107
  75. brainstate/nn/_ltp.py +28 -28
  76. brainstate/nn/_module.py +377 -377
  77. brainstate/nn/_module_test.py +40 -40
  78. brainstate/nn/_neuron.py +705 -705
  79. brainstate/nn/_neuron_test.py +161 -161
  80. brainstate/nn/_normalizations.py +975 -918
  81. brainstate/nn/_normalizations_test.py +73 -73
  82. brainstate/{functional → nn}/_others.py +46 -46
  83. brainstate/nn/_poolings.py +1177 -1177
  84. brainstate/nn/_poolings_test.py +217 -217
  85. brainstate/nn/_projection.py +486 -486
  86. brainstate/nn/_rate_rnns.py +554 -554
  87. brainstate/nn/_rate_rnns_test.py +63 -63
  88. brainstate/nn/_readout.py +209 -209
  89. brainstate/nn/_readout_test.py +53 -53
  90. brainstate/nn/_stp.py +236 -236
  91. brainstate/nn/_synapse.py +505 -505
  92. brainstate/nn/_synapse_test.py +131 -131
  93. brainstate/nn/_synaptic_projection.py +423 -423
  94. brainstate/nn/_synouts.py +162 -162
  95. brainstate/nn/_synouts_test.py +57 -57
  96. brainstate/nn/_utils.py +89 -89
  97. brainstate/nn/metrics.py +388 -388
  98. brainstate/optim/__init__.py +38 -38
  99. brainstate/optim/_base.py +64 -64
  100. brainstate/optim/_lr_scheduler.py +448 -448
  101. brainstate/optim/_lr_scheduler_test.py +50 -50
  102. brainstate/optim/_optax_optimizer.py +152 -152
  103. brainstate/optim/_optax_optimizer_test.py +53 -53
  104. brainstate/optim/_sgd_optimizer.py +1104 -1104
  105. brainstate/random/__init__.py +24 -24
  106. brainstate/random/_rand_funs.py +3616 -3616
  107. brainstate/random/_rand_funs_test.py +567 -567
  108. brainstate/random/_rand_seed.py +210 -210
  109. brainstate/random/_rand_seed_test.py +48 -48
  110. brainstate/random/_rand_state.py +1409 -1409
  111. brainstate/random/_random_for_unit.py +52 -52
  112. brainstate/surrogate.py +1957 -1957
  113. brainstate/transform.py +23 -23
  114. brainstate/typing.py +304 -304
  115. brainstate/util/__init__.py +50 -50
  116. brainstate/util/caller.py +98 -98
  117. brainstate/util/error.py +55 -55
  118. brainstate/util/filter.py +469 -469
  119. brainstate/util/others.py +540 -540
  120. brainstate/util/pretty_pytree.py +945 -945
  121. brainstate/util/pretty_pytree_test.py +159 -159
  122. brainstate/util/pretty_repr.py +328 -328
  123. brainstate/util/pretty_table.py +2954 -2954
  124. brainstate/util/scaling.py +258 -258
  125. brainstate/util/struct.py +523 -523
  126. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info}/METADATA +91 -99
  127. brainstate-0.1.10.dist-info/RECORD +130 -0
  128. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info}/WHEEL +1 -1
  129. {brainstate-0.1.8.dist-info → brainstate-0.1.10.dist-info/licenses}/LICENSE +202 -202
  130. brainstate/functional/_normalization.py +0 -81
  131. brainstate/functional/_spikes.py +0 -204
  132. brainstate-0.1.8.dist-info/RECORD +0 -132
  133. {brainstate-0.1.8.dist-info → brainstate-0.1.10.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 brainstate as bc
19
-
20
-
21
- class TestMixin(unittest.TestCase):
22
- def test_mixin(self):
23
- self.assertTrue(bc.mixin.Mixin)
24
- self.assertTrue(bc.mixin.ParamDesc)
25
- self.assertTrue(bc.mixin.ParamDescriber)
26
- self.assertTrue(bc.mixin.JointTypes)
27
- self.assertTrue(bc.mixin.OneOfTypes)
28
- self.assertTrue(bc.mixin.Mode)
29
- self.assertTrue(bc.mixin.Batching)
30
- self.assertTrue(bc.mixin.Training)
31
-
32
-
33
- class TestMode(unittest.TestCase):
34
- def test_JointMode(self):
35
- a = bc.mixin.JointMode(bc.mixin.Batching(), bc.mixin.Training())
36
- self.assertTrue(a.is_a(bc.mixin.JointTypes[bc.mixin.Batching, bc.mixin.Training]))
37
- self.assertTrue(a.has(bc.mixin.Batching))
38
- self.assertTrue(a.has(bc.mixin.Training))
39
- b = bc.mixin.JointMode(bc.mixin.Batching())
40
- self.assertTrue(b.is_a(bc.mixin.JointTypes[bc.mixin.Batching]))
41
- self.assertTrue(b.is_a(bc.mixin.Batching))
42
- self.assertTrue(b.has(bc.mixin.Batching))
43
-
44
- def test_Training(self):
45
- a = bc.mixin.Training()
46
- self.assertTrue(a.is_a(bc.mixin.Training))
47
- self.assertTrue(a.is_a(bc.mixin.JointTypes[bc.mixin.Training]))
48
- self.assertTrue(a.has(bc.mixin.Training))
49
- self.assertTrue(a.has(bc.mixin.JointTypes[bc.mixin.Training]))
50
- self.assertFalse(a.is_a(bc.mixin.Batching))
51
- self.assertFalse(a.has(bc.mixin.Batching))
52
-
53
- def test_Batching(self):
54
- a = bc.mixin.Batching()
55
- self.assertTrue(a.is_a(bc.mixin.Batching))
56
- self.assertTrue(a.is_a(bc.mixin.JointTypes[bc.mixin.Batching]))
57
- self.assertTrue(a.has(bc.mixin.Batching))
58
- self.assertTrue(a.has(bc.mixin.JointTypes[bc.mixin.Batching]))
59
-
60
- self.assertFalse(a.is_a(bc.mixin.Training))
61
- self.assertFalse(a.has(bc.mixin.Training))
62
-
63
- def test_Mode(self):
64
- a = bc.mixin.Mode()
65
- self.assertTrue(a.is_a(bc.mixin.Mode))
66
- self.assertTrue(a.is_a(bc.mixin.JointTypes[bc.mixin.Mode]))
67
- self.assertTrue(a.has(bc.mixin.Mode))
68
- self.assertTrue(a.has(bc.mixin.JointTypes[bc.mixin.Mode]))
69
-
70
- self.assertFalse(a.is_a(bc.mixin.Training))
71
- self.assertFalse(a.has(bc.mixin.Training))
72
- self.assertFalse(a.is_a(bc.mixin.Batching))
73
- self.assertFalse(a.has(bc.mixin.Batching))
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 ._poolings import *
53
- from ._poolings import __all__ as poolings_all
54
- from ._projection import *
55
- from ._projection import __all__ as projection_all
56
- from ._rate_rnns import *
57
- from ._rate_rnns import __all__ as rate_rnns
58
- from ._readout import *
59
- from ._readout import __all__ as readout_all
60
- from ._stp import *
61
- from ._stp import __all__ as stp_all
62
- from ._synapse import *
63
- from ._synapse import __all__ as dyn_synapse_all
64
- from ._synaptic_projection import *
65
- from ._synaptic_projection import __all__ as _syn_proj_all
66
- from ._synouts import *
67
- from ._synouts import __all__ as synouts_all
68
- from ._utils import *
69
- from ._utils import __all__ as utils_all
70
-
71
- __all__ = (
72
- [
73
- 'metrics',
74
- ]
75
- + collective_ops_all
76
- + common_all
77
- + elementwise_all
78
- + module_all
79
- + exp_euler_all
80
- + utils_all
81
- + dyn_all
82
- + projection_all
83
- + state_delay_all
84
- + synouts_all
85
- + conv_all
86
- + linear_all
87
- + normalizations_all
88
- + poolings_all
89
- + fixedprob_all
90
- + linear_mv_all
91
- + embed_all
92
- + dropout_all
93
- + elementwise_all
94
- + dyn_neuron_all
95
- + dyn_synapse_all
96
- + inputs_all
97
- + rate_rnns
98
- + readout_all
99
- + stp_all
100
- + ltp_all
101
- + _syn_proj_all
102
- )
103
-
104
- del (
105
- collective_ops_all,
106
- common_all,
107
- module_all,
108
- exp_euler_all,
109
- utils_all,
110
- dyn_all,
111
- projection_all,
112
- state_delay_all,
113
- synouts_all,
114
- conv_all,
115
- linear_all,
116
- normalizations_all,
117
- poolings_all,
118
- embed_all,
119
- fixedprob_all,
120
- linear_mv_all,
121
- dropout_all,
122
- elementwise_all,
123
- dyn_neuron_all,
124
- dyn_synapse_all,
125
- inputs_all,
126
- readout_all,
127
- rate_rnns,
128
- stp_all,
129
- ltp_all,
130
- _syn_proj_all,
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
+ )