MkldnnMaxPool2d =============== Applies max pooling 2D (MKLDNN). **Abstract Signature:** ``MkldnnMaxPool2d(input: Tensor, kernel_size: int, stride: int, padding: int = 0, dilation: int = 1, ceil_mode: bool = False)`` .. raw:: html
jax.lax.reduce_window