Hi, can you show the `critical schedule statement` that makes the conv2d op fused into its following elementwise op, e.g. topi.nn.relu?
I found it should use `compute_at` but it seems to be a little hard to use that well. ``` B = conv2d(A) C = topi.nn.relu(B) .. schedule_for_conv2d(s[B], ..) s[B].compute_at(s[C], C.op.axis[what is reasonable?]) ``` --- [Visit Topic](https://discuss.tvm.ai/t/how-to-schedule-fused-ops/2522/9) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/a4c97bef95ec8ce7f477d864461ff404db22a864b1b001be253063172e4c94fc).