code: def attn_qkv():
      # TransfoXL's qkv net is reversed.
      return [
          Layer(
              weight="dec_attn.qkv_net.weight",
              n_fused=3,
              reversed=True,
          ),
          Layer(weight="dec_attn.r_net.weight"),
          Layer(bias="dec_attn.r_r_bias"),
          Layer(bias="dec_attn.r_w_bias"),
      ]
  
file_path: parallelformers/parallelformers/policies/transfo_xl.py
imports: from transformers.models.transfo_xl.modeling_transfo_xl import RelPartialLearnableDecoderLayer
  from parallelformers.policies.base import Layer, Policy
  from parallelformers.utils.dist_utils import AllReduceLinear
label_desc: Use image node_id 2 for calling the TransfoXLPolicy obj's underlying member method code with example usage: obj.attn_qkv() and returns: list