Searched refs:conv_in_channel (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_reshape.py | 532 def __init__(self, conv_in_channel, conv_out_channel, argument
535 self.conv = nn.Conv2d(in_channels=conv_in_channel, out_channels=conv_out_channel,
574 net = ParallelReduceMeanNet(conv_in_channel=3, conv_out_channel=64, reducemean_axis=(2, 3),
594 net = ParallelReduceMeanNet(conv_in_channel=3, conv_out_channel=64, reducemean_axis=(2, 3),
665 net = ParallelReduceMeanNet(conv_in_channel=3, conv_out_channel=64, reducemean_keep_dims=True,
|