def METHOD_NAME(attrs, inputs, _):
    """Compute definition of roi_pool"""
    assert attrs.layout == "NCHW", "only support nchw for now"
    return [
        topi.vision.rcnn.roi_pool_nchw(
            inputs[0],
            inputs[1],
            pooled_size=get_const_tuple(attrs.pooled_size),
            spatial_scale=attrs.spatial_scale,
        )
    ]