Searched refs:MakeLayer3 (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/ut/python/exec/ |
| D | resnet_example.py | 112 class MakeLayer3(nn.Cell): class
118 super(MakeLayer3, self).__init__()
195 self.layer1 = MakeLayer3(
201 self.layer4 = MakeLayer3(
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | resnet.py | 225 class MakeLayer3(nn.Cell): class
228 super(MakeLayer3, self).__init__()
257 self.layer4 = MakeLayer3(block, in_channels=1024, out_channels=2048, stride=2)
|
| /third_party/mindspore/tests/st/networks/models/ |
| D | resnetv1_5.py | 240 class MakeLayer3(nn.Cell): class
243 super(MakeLayer3, self).__init__()
272 self.layer4 = MakeLayer3(block, in_channels=1024, out_channels=2048, stride=2)
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet.py | 244 class MakeLayer3(nn.Cell): class
247 super(MakeLayer3, self).__init__()
276 self.layer4 = MakeLayer3(block, in_channels=1024, out_channels=2048, stride=2)
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_resnet_air.py | 263 class MakeLayer3(Cell): class
266 super(MakeLayer3, self).__init__()
297 self.layer4 = MakeLayer3(
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_resnet.py | 269 class MakeLayer3(Cell): class
272 super(MakeLayer3, self).__init__()
303 self.layer4 = MakeLayer3(
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_resnet50_gpu.py | 248 class MakeLayer3(nn.Cell): class
251 super(MakeLayer3, self).__init__()
280 self.layer4 = MakeLayer3(block, in_channels=1024, out_channels=2048, stride=2)
|
| D | test_pynative_resnet50_ascend.py | 286 class MakeLayer3(nn.Cell): class
289 super(MakeLayer3, self).__init__()
318 self.layer4 = MakeLayer3(block, in_channels=1024, out_channels=2048, stride=2)
|