Searched refs:ConvNet (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/tests/ut/python/pynative_mode/nn/ |
D | test_cell.py | 86 class ConvNet(nn.Cell): class 93 super(ConvNet, self).__init__() 94 … self.conv1 = nn.Conv2d(3, ConvNet.output_ch, kernel_size=7, stride=2, pad_mode='pad', padding=3) 95 self.bn1 = nn.BatchNorm2d(ConvNet.output_ch) 100 int(ConvNet.image_h * ConvNet.image_w * ConvNet.output_ch / (4 * 4)), 304 net = ConvNet()
|
/third_party/mindspore/tests/ut/python/nn/ |
D | test_cell.py | 79 class ConvNet(nn.Cell): class 86 super(ConvNet, self).__init__() 87 … self.conv1 = nn.Conv2d(3, ConvNet.output_ch, kernel_size=7, stride=2, pad_mode="pad", padding=3) 88 self.bn1 = nn.BatchNorm2d(ConvNet.output_ch) 93 int(ConvNet.image_h * ConvNet.image_w * ConvNet.output_ch / (4 * 4)), 204 net = ConvNet()
|
/third_party/mindspore/tests/ut/python/pynative_mode/ |
D | test_ms_function.py | 48 class ConvNet(nn.Cell): class 50 super(ConvNet, self).__init__() 73 net = ConvNet()
|
D | test_framstruct.py | 309 class ConvNet(nn.Cell): class 311 super(ConvNet, self).__init__() 328 conv = ConvNet()
|