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/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/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/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)
|