code: def compute_workload(self):
      if self.mode == "fwd":
          count = 1
      else:
          count = 1 + (1 + 1)
  
      op_count = 2 * self.B * self.M * self.N * self.K
  
      return op_count * count
  
file_path: pytorch/benchmarks/tensorexpr/matmul.py
imports: import numpy
  from .None import benchmark
label_desc: Use image node_id 7 for calling the MatMulBench obj's underlying member method code with example usage: obj.compute_workload() and returns: unknown