fn
adaptive_max_pool2d_op
→TensorImplPtrint adaptive_max_pool2d_op(const int & x, int OH, int OW)Python wrappers
Public Python APIs implemented by this engine symbol.Two-dimensional adaptive max-pool to a fixed (OH, OW) grid.
Used in detection / segmentation models that must produce a fixed spatial grid regardless of input resolution (e.g. RoIAlign-style heads, classification networks fine-tuned on multi-scale data).
Parameters
xTensorImplPtrInput of shape
(B, C, H, W).OH, OWintTarget output extents. Must satisfy
H % OH == 0 and W % OW == 0.Returns
TensorImplPtrOutput of shape (B, C, OH, OW).
Raises
ShapeMismatchIf
x is not 4-D.not_implementedIf any spatial axis is not evenly divisible by its target.