code: def compute_workload(self):
      if self.mode == "fwd":
          count = 1
      elif self.mode == "both":
          count = 1 + (1 + 1)
      else:
          raise ValueError(f"invalid mode: {self.mode}")
  
      op_count = (
          self.N
          * self.iC
          / self.groups
          * self.oC
          * self.kernel_size
          * self.kernel_size
          * self.H
          * self.W
      )
      op_count *= 2
  
      return op_count * count
  
file_path: pytorch/benchmarks/tensorexpr/conv.py
imports: from .None import benchmark
label_desc: Use image node_id 5 for calling the ConvImplBench obj's underlying member method code with example usage: obj.compute_workload() and returns: unknown