| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | darknet.py | 99 layer_nums, argument
108 if not len(layer_nums) == len(in_channels) == len(out_channels) == 5:
136 layer_nums[0],
140 layer_nums[1],
144 layer_nums[2],
148 layer_nums[3],
152 layer_nums[4],
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | resnet.py | 158 layer_nums, argument
165 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
174 layer_nums[0],
179 layer_nums[1],
184 layer_nums[2],
189 layer_nums[3],
|
| /third_party/mindspore/tests/ut/python/model/ |
| D | resnet.py | 158 layer_nums, argument
165 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
174 layer_nums[0],
179 layer_nums[1],
184 layer_nums[2],
189 layer_nums[3],
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src/ |
| D | resnet.py | 158 layer_nums, argument
165 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
174 layer_nums[0],
179 layer_nums[1],
184 layer_nums[2],
189 layer_nums[3],
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/ |
| D | resnet.py | 72 layer_nums, argument
78 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
93 layer_nums[0],
100 layer_nums[1],
107 layer_nums[2],
114 layer_nums[3],
|
| D | resnet50v1.py | 72 layer_nums, argument
78 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
93 layer_nums[0],
100 layer_nums[1],
107 layer_nums[2],
114 layer_nums[3],
|
| /third_party/mindspore/tests/st/quantization/resnet50_quant/ |
| D | resnet_quant_manual.py | 202 layer_nums, argument
209 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
216 layer_nums[0],
221 layer_nums[1],
226 layer_nums[2],
231 layer_nums[3],
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | resnet.py | 218 layer_nums, argument
226 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
245 layer_nums[0],
251 layer_nums[1],
257 layer_nums[2],
264 layer_nums[3],
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_resnet_sharding_propagation2.py | 133 layer_nums, argument
143 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
153 layer_nums[0],
158 layer_nums[1],
163 layer_nums[2],
168 layer_nums[3],
|
| D | test_auto_parallel_resnet_sharding_propagation.py | 134 layer_nums, argument
144 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
154 layer_nums[0],
159 layer_nums[1],
164 layer_nums[2],
169 layer_nums[3],
|
| D | test_auto_parallel_resnet.py | 137 layer_nums, argument
145 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
156 layer_nums[0],
161 layer_nums[1],
166 layer_nums[2],
171 layer_nums[3],
|
| D | test_auto_parallel_transformer.py | 91 def __init__(self, layer_nums=1): argument
93 self.layer = self._make_layer(layer_nums)
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | resnet50_expand_loss.py | 162 layer_nums, argument
171 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
181 layer_nums[0],
186 layer_nums[1],
191 layer_nums[2],
196 layer_nums[3],
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 155 layer_nums, argument
162 if not len(layer_nums) == len(in_channels) == len(out_channels) == 4:
172 layer_nums[0],
177 layer_nums[1],
182 layer_nums[2],
187 layer_nums[3],
|