Home
last modified time | relevance | path

Searched refs:test_max_pool2d_nhwc (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_pooling.py1079 def test_max_pool2d_nhwc(self, device, dtype): member in TestPoolingNNDeviceType
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py1564 def test_max_pool2d_nhwc(self, X, kernel, stride, dilation, padding, ceil_mode): member in TestQuantizedOps