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 
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 

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 

Reply via email to