Hi, @gkolhe Unless you want to apply specific schedule in TOPI, default schedule should work.
R = topi.nn.batch_matmul(A, B)
s = te.create_schedule(R.op)
return tvm.build(s, [A, B, R], name=name)Hi, @gkolhe Unless you want to apply specific schedule in TOPI, default schedule should work.
R = topi.nn.batch_matmul(A, B)
s = te.create_schedule(R.op)
return tvm.build(s, [A, B, R], name=name)