@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).