e.g
Blockquote A_shape=[i,j,k, l] A = tvm.placeholder(A_shape,dtype=“float16”, name=“A”) B_shape=[i, j, kl, 1] B = tvm.compute(B_shape, …) s = tvm.create_schedule([B.op])
s[B].op.axis[2] is equal s[A].op.axis[2]*s[A].op.axis[3] can i get those information with stage?