Searched refs:MakeLayer0 (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_operator_model_parallel.py | 277 class MakeLayer0(Cell): class
280 super(MakeLayer0, self).__init__()
301 self.layer1 = MakeLayer0(
324 self.layer1 = MakeLayer0(
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | resnet.py | 169 class MakeLayer0(nn.Cell): class
172 super(MakeLayer0, self).__init__()
254 self.layer1 = MakeLayer0(block, in_channels=64, out_channels=256, stride=1)
|
| /third_party/mindspore/tests/st/networks/models/ |
| D | resnetv1_5.py | 184 class MakeLayer0(nn.Cell): class
187 super(MakeLayer0, self).__init__()
269 self.layer1 = MakeLayer0(block, in_channels=64, out_channels=256, stride=1)
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet.py | 188 class MakeLayer0(nn.Cell): class
191 super(MakeLayer0, self).__init__()
273 self.layer1 = MakeLayer0(block, in_channels=64, out_channels=256, stride=1)
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_resnet_air.py | 204 class MakeLayer0(Cell): class
207 super(MakeLayer0, self).__init__()
291 self.layer1 = MakeLayer0(
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_resnet.py | 210 class MakeLayer0(Cell): class
213 super(MakeLayer0, self).__init__()
297 self.layer1 = MakeLayer0(
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_resnet50_gpu.py | 192 class MakeLayer0(nn.Cell): class
195 super(MakeLayer0, self).__init__()
277 self.layer1 = MakeLayer0(block, in_channels=64, out_channels=256, stride=1)
|
| D | test_pynative_resnet50_ascend.py | 230 class MakeLayer0(nn.Cell): class
233 super(MakeLayer0, self).__init__()
315 self.layer1 = MakeLayer0(block, in_channels=64, out_channels=256, stride=1)
|