[Bug] [TOPI]te_create_prime cannot generate valid tir for the topk with ret_type='both'

### Actual behavior

```
# from tvm.script import ir as I
# from tvm.script import tir as T

@I.ir_module
class Module:
@T.prim_func
def main(data: T.Buffer((128, 64), "float32")):
T.func_attr({"tir.noalias": T.bool(True)})
T.evaluate(0)
```

### Environment

tvm-0.21.dev0

### Steps to reproduce

```
import tvm
from tvm import te, topi, tir

data = te.placeholder((128, 64), dtype='float32', name='data')
op_output = topi.topk(data, k=10, axis=1, ret_type='both')
print(op_output)
sch = tir.Schedule(te.create_prim_func([data, op_output]))
print(sch.mod)
```

### Triage

* needs-triage
* te

Reproduce the issue with topi.topk(..., ret_type='both'), then inspect te.create_prim_func and the resulting tir.Schedule module. Trace why this combination produces invalid TIR; done means the same reproduction generates valid TIR for the topk outputs.

Written by the indexing model from the issue text.