code: def set_attention(self, qkvw, qkvb, dense_w, dense_b):
      super().set_attention(qkvw, qkvb, dense_w, dense_b)
      self.set_q_k_v()
  
file_path: DeepSpeed/deepspeed/module_inject/containers/features/split_qkv.py
imports: from abc import abstractmethod
  import torch
  from .hybrid_engine import HybridEngineContainer
label_desc: Use image node_id 1 for calling the HybridSplitQKVContainer obj's underlying member method code with example usage: obj.set_attention(qkvw, qkvb, dense_w, dense_b) without return types