def test_2d_max_shape(self):
        input_param = {'func': 'CONSTANT',
                       'kernel_size': [1, 3],
                       'stride': 3}
        self._test_nd_downsample_output_shape(rank=2,
                                              param_dict=input_param,
                                              output_shape=(2, 6, 6, 8))