Home
last modified time | relevance | path

Searched refs:bn_with_initialize (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/tests/st/tbe_networks/
Dresnet.py53 def bn_with_initialize(out_channels): function
88 self.bn1 = bn_with_initialize(out_chls)
91 self.bn2 = bn_with_initialize(out_chls)
131 self.bn1 = bn_with_initialize(out_chls)
134 self.bn2 = bn_with_initialize(out_chls)
143 self.bn_down_sample = bn_with_initialize(out_channels)
250 self.bn1 = bn_with_initialize(64)
/third_party/mindspore/tests/ut/python/exec/
Dresnet_example.py40 def bn_with_initialize(out_channels): function
68 self.bn1 = bn_with_initialize(out_chls)
71 self.bn2 = bn_with_initialize(out_chls)
74 self.bn3 = bn_with_initialize(out_channels)
81 self.bn_down_sample = bn_with_initialize(out_channels)
191 self.bn1 = bn_with_initialize(64)
/third_party/mindspore/tests/st/networks/models/
Dresnetv1_5.py61 def bn_with_initialize(out_channels): function
103 self.bn1 = bn_with_initialize(out_chls)
106 self.bn2 = bn_with_initialize(out_chls)
146 self.bn1 = bn_with_initialize(out_chls)
149 self.bn2 = bn_with_initialize(out_chls)
158 self.bn_down_sample = bn_with_initialize(out_channels)
265 self.bn1 = bn_with_initialize(64)
/third_party/mindspore/tests/st/mem_reuse/
Dresnet.py66 def bn_with_initialize(out_channels): function
107 self.bn1 = bn_with_initialize(out_chls)
110 self.bn2 = bn_with_initialize(out_chls)
150 self.bn1 = bn_with_initialize(out_chls)
153 self.bn2 = bn_with_initialize(out_chls)
162 self.bn_down_sample = bn_with_initialize(out_channels)
269 self.bn1 = bn_with_initialize(64)
/third_party/mindspore/tests/st/export_and_load/
Dtest_resnet_air.py79 def bn_with_initialize(out_channels): function
122 self.bn1 = bn_with_initialize(out_chls)
125 self.bn2 = bn_with_initialize(out_chls)
165 self.bn1 = bn_with_initialize(out_chls)
168 self.bn2 = bn_with_initialize(out_chls)
178 self.bn_down_sample = bn_with_initialize(out_channels)
287 self.bn1 = bn_with_initialize(64)
/third_party/mindspore/tests/st/networks/
Dtest_gpu_resnet.py85 def bn_with_initialize(out_channels): function
128 self.bn1 = bn_with_initialize(out_chls)
131 self.bn2 = bn_with_initialize(out_chls)
171 self.bn1 = bn_with_initialize(out_chls)
174 self.bn2 = bn_with_initialize(out_chls)
184 self.bn_down_sample = bn_with_initialize(out_channels)
293 self.bn1 = bn_with_initialize(64)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_operator_model_parallel.py147 def bn_with_initialize(out_channels): function
192 self.bn1 = bn_with_initialize(out_chls)
195 self.bn2 = bn_with_initialize(out_chls)
237 self.bn1 = bn_with_initialize(out_chls)
240 self.bn2 = bn_with_initialize(out_chls)
251 self.bn_down_sample = bn_with_initialize(out_channels)
298 self.bn1 = bn_with_initialize(64)
Dtest_auto_parallel_BN_PReLU.py40 def bn_with_initialize(out_channels): function
60 self.bn = bn_with_initialize(16)
Dtest_auto_parallel_tuple_depend.py51 def bn_with_initialize(out_channels): function
65 self.bn1 = bn_with_initialize(2048)
Dtest_batchnorm_batch_parallel.py81 def bn_with_initialize(out_channels): function
99 self.bn1 = bn_with_initialize(64)
Dtest_one_hot_net.py244 def bn_with_initialize(out_channels): function
256 self.batch_norm = bn_with_initialize(2)
Dtest_reshape.py493 def bn_with_initialize(out_channels): function
505 self.batch_norm = bn_with_initialize(2)
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_resnet50_gpu.py76 def bn_with_initialize(out_channels): function
111 self.bn1 = bn_with_initialize(out_chls)
114 self.bn2 = bn_with_initialize(out_chls)
154 self.bn1 = bn_with_initialize(out_chls)
157 self.bn2 = bn_with_initialize(out_chls)
166 self.bn_down_sample = bn_with_initialize(out_channels)
273 self.bn1 = bn_with_initialize(64)
Dtest_pynative_resnet50_ascend.py114 def bn_with_initialize(out_channels): function
149 self.bn1 = bn_with_initialize(out_chls)
152 self.bn2 = bn_with_initialize(out_chls)
192 self.bn1 = bn_with_initialize(out_chls)
195 self.bn2 = bn_with_initialize(out_chls)
204 self.bn_down_sample = bn_with_initialize(out_channels)
311 self.bn1 = bn_with_initialize(64)