Searched refs:AvgPool2d (Results 1 – 21 of 21) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_avgpool_op.py | 30 net = nn.AvgPool2d(kernel_size=2, stride=1, pad_mode='valid')
48 net = nn.AvgPool2d(kernel_size=2, stride=2, pad_mode='valid')
64 net = nn.AvgPool2d(kernel_size=3, stride=2, pad_mode='valid')
79 net = nn.AvgPool2d(kernel_size=3, stride=2, pad_mode='same')
|
| D | test_avgpool_grad_op.py | 43 net = nn.AvgPool2d(kernel_size=3, stride=2, pad_mode='valid')
|
| /third_party/mindspore/tests/ut/python/pynative_mode/nn/ |
| D | test_pooling.py | 29 avg_pool = nn.AvgPool2d(kernel_size, stride)
43 nn.AvgPool2d(kernel_size, stride)
|
| D | test_container.py | 32 avg_pool = nn.AvgPool2d(kernel_size, stride)
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_celllist.py | 20 from mindspore.nn import AvgPool2d
55 self.seq = SequentialCell([AvgPool2d(3, 1), ReLU(), Flatten()])
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_pooling.py | 40 self.avgpool = nn.AvgPool2d(kernel_size,
|
| D | resnet_example.py | 204 self.avgpool = nn.AvgPool2d(7, 1)
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ge/ops/ |
| D | test_pooling.py | 30 avg_pool = nn.AvgPool2d(kernel_size, stride)
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | pooling.py | 228 class AvgPool2d(_PoolNd): class
297 super(AvgPool2d, self).__init__(kernel_size, stride, pad_mode, data_format)
|
| D | image.py | 26 from .pooling import AvgPool2d
373 self.avg_pool = AvgPool2d(kernel_size=2, stride=2, pad_mode='valid')
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_pooling.py | 30 self.avgpool = nn.AvgPool2d(kernel_size, stride)
|
| D | test_container.py | 32 avg_pool = nn.AvgPool2d(kernel_size, stride)
|
| /third_party/mindspore/tests/ut/python/model/ |
| D | res18_example.py | 120 self.avgpool = nn.AvgPool2d(7, 1)
189 self.avgpool = nn.AvgPool2d(7, 1)
|
| /third_party/mindspore/tests/ut/python/communication/ |
| D | test_data_parallel_resnet.py | 122 self.avgpool = nn.AvgPool2d(7, 1)
191 self.avgpool = nn.AvgPool2d(7, 1)
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | NetworkInNetwork.py | 59 nn.AvgPool2d(kernel_size=8, stride=1, pad_mode='valid')
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops_check.py | 167 'block': nn.AvgPool2d(5, pad_mode='same'),
170 'block': nn.AvgPool2d(5, pad_mode='valid'),
|
| /third_party/mindspore/tests/perf_test/ |
| D | resnet_example.py | 117 self.avgpool = nn.AvgPool2d(7, 1)
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
| D | deeplabv3.py | 105 … self.global_pooling = nn.AvgPool2d(kernel_size=(int(feature_shape[0]), int(feature_shape[1])))
110 self.global_poolings.append(nn.AvgPool2d(kernel_size=(int(pooling_h), int(pooling_w))))
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_debug_location.py | 154 self.seq = nn.SequentialCell([nn.AvgPool2d(3, 1), nn.ReLU(), nn.Flatten()])
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_resnet_air.py | 300 self.pool = nn.AvgPool2d(7, 1)
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_resnet.py | 306 self.pool = nn.AvgPool2d(7, 1)
|