b7e0cfb6d469c3745ae2195908daadea9c64d87e,python/tvm/relay/op/strategy/generic.py,,roi_align_strategy,#Any#Any#Any#Any#,1060

Before Change

    strategy = _op.OpStrategy()
    layout = attrs.layout
    assert layout == "NCHW", "only support nchw for now"
    strategy.add_implementation(
        wrap_compute_roi_align(topi.vision.rcnn.roi_align_nchw),
        wrap_topi_schedule(topi.generic.schedule_roi_align),
        name="roi_align.generic",
    )
    return strategy


// sparse_fill_empty_rows

After Change

    roi_align generic strategy
    strategy = _op.OpStrategy()
    layout = attrs.layout
    if layout == "NCHW":
        strategy.add_implementation(
            wrap_compute_roi_align(topi.vision.rcnn.roi_align_nchw),
            wrap_topi_schedule(topi.generic.schedule_roi_align),
            name="roi_align.generic",
        )
    else:
        assert layout == "NHWC", "layout must be NCHW or NHWC."
        strategy.add_implementation(
            wrap_compute_roi_align(topi.vision.rcnn.roi_align_nhwc),
            wrap_topi_schedule(topi.generic.schedule_roi_align),
            name="roi_align.generic",
        )
    return strategy


// sparse_fill_empty_rows

Instances