Home
last modified time | relevance | path

Searched refs:ConvNet (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/ut/python/pynative_mode/nn/
Dtest_cell.py86 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/
Dtest_cell.py79 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/
Dtest_ms_function.py48 class ConvNet(nn.Cell): class
50 super(ConvNet, self).__init__()
73 net = ConvNet()
Dtest_framstruct.py309 class ConvNet(nn.Cell): class
311 super(ConvNet, self).__init__()
328 conv = ConvNet()