code: def attn_qkv():
      return [
          Layer(weight="attn.q_proj.weight"),
          Layer(weight="attn.k_proj.weight"),
          Layer(weight="attn.v_proj.weight"),
      ]
  
file_path: parallelformers/parallelformers/policies/gptj.py
imports: from transformers.models.gptj.modeling_gptj import GPTJBlock
  from parallelformers.policies.base import Layer, Policy
  from parallelformers.utils import AllReduceLinear
label_desc: Use image node_id 2 for calling the GPTJPolicy obj's underlying member method code with example usage: obj.attn_qkv() and returns: list