[Unity] Update docs for operators

[Ubospica]

This PR:

• updates docs for several operators,
• add an 1 default value for the axis parameter of batch_norm,
• rename module tvm.relax.transform.legalize_ops.creation to tvm.relax.transform.legalize_ops.create, which is align with tvm.relax.op.create, and
• fixes a previous upstream error in tests/python/relax/test_transform_legalize_ops_grad.py

[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 @quic-sanirudh _{See #10317 for details}

_{Generated by tvm-bot}