In this case, you can it in the op alter layout. You can check out the example from https://github.com/apache/incubator-tvm/blob/master/python/tvm/topi/x86/conv2d_alter_op.py#L40.
In this case, you can it in the op alter layout. You can check out the example from https://github.com/apache/incubator-tvm/blob/master/python/tvm/topi/x86/conv2d_alter_op.py#L40.