[TIR] Allow TransformLayout index_map to contain RVs

[vinx13]

This changes index_map argument of TransformLayout from attr to input. It allows the index map to capture the random variables in the environment. This will allow tuning with different index_map applied.

cc @Hzfengsy

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @Hzfengsy, @junrushao, @quic-sanirudh, @shingjan _{See #10317 for details}

_{Generated by tvm-bot}