About TensorIR schedule, how can I get the reduce loop?

Questions
1

image
image666×580 36.2 KB
I want to get the block’s reduce loop corresponding to the reduce axis. for example, the reduce loop is i1_0_1. how can I get the loop rv of it ?

2

you can get the block block = s.get_block('B') and then s.get_loops(block)[-1] to get the innermost loop