code: def is_qkv(self, data):
      return (self.mp_size * data.shape[0]) / data.shape[1] == 3 or (
          self.mp_size * data.shape[1]
      ) / data.shape[0] == 3
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/runtime/weight_quantizer.py
imports: import torch
  from ..module_inject.replace_policy import HFBertLayerPolicy, replace_policies
label_desc: Use image node_id 4 for calling the WeightQuantization obj's underlying member method code with example usage: obj.is_qkv(data) and returns: unknown