@giuseros @anijain2305 Let us accept one more argument like `alter_op_layout`  

```
@tvm.target.generic_func
def conv2d_alter_layout(attrs, inputs, tinfos, out_type):

@tvm.target.generic_func
def qnn_conv2d_legalize(attrs, inputs, types):
    """Default legalization is None."""
    return None
```
Then we could leverage `relay.backend.compile_engine.select_implementation`





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/quantized-models-and-legalization-pass/8253/8)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/dd316de373aecf2859b493c070c365411dab1b41987380112e6995b01c52da90).

Reply via email to