MaxRoiPool

Description

Consumes an input tensor X and region of interests (RoIs) to apply max pooling across each RoI, to produce output 4-D tensor of shape (num_rois, channels, pooled_shape[0], pooled_shape[1]).

Input

• x: input tensor of type float16.
• rois: input tensor of type float16.

Attribute

• pooled_shape: list of ints.
• spatial_scale: float. The default value is 1.0.

Output

y: output tensor of type float16. The shape is a 4-D tensor. Tensor format: (num_roi, channels, pooled_shape[0], pooled_shape[1]).

Constraints

The atc tool does not support the input of the float type when the --precision_mode=must_keep_origin_dtype parameter is used.

ONNX Opset Support

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16