Searched refs:MakeLayer1 (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | resnet.py | 185 class MakeLayer1(nn.Cell): class
188 super(MakeLayer1, self).__init__()
255 self.layer2 = MakeLayer1(block, in_channels=256, out_channels=512, stride=2)
|
| /third_party/mindspore/tests/st/networks/models/ |
| D | resnetv1_5.py | 200 class MakeLayer1(nn.Cell): class
203 super(MakeLayer1, self).__init__()
270 self.layer2 = MakeLayer1(block, in_channels=256, out_channels=512, stride=2)
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet.py | 204 class MakeLayer1(nn.Cell): class
207 super(MakeLayer1, self).__init__()
274 self.layer2 = MakeLayer1(block, in_channels=256, out_channels=512, stride=2)
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_resnet_air.py | 221 class MakeLayer1(Cell): class
224 super(MakeLayer1, self).__init__()
293 self.layer2 = MakeLayer1(
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_resnet.py | 227 class MakeLayer1(Cell): class
230 super(MakeLayer1, self).__init__()
299 self.layer2 = MakeLayer1(
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_resnet50_gpu.py | 208 class MakeLayer1(nn.Cell): class
211 super(MakeLayer1, self).__init__()
278 self.layer2 = MakeLayer1(block, in_channels=256, out_channels=512, stride=2)
|
| D | test_pynative_resnet50_ascend.py | 246 class MakeLayer1(nn.Cell): class
249 super(MakeLayer1, self).__init__()
316 self.layer2 = MakeLayer1(block, in_channels=256, out_channels=512, stride=2)
|