code: def init(self, M, N, K, dtype):
      self.input_one = self.tensor([M, N, K], dtype)
      self.input_two = self.tensor([M, N, K], dtype)
      self.output = self.tensor([M, N, K], dtype)
      self.set_module_name("add")
  
file_path: pytorch/benchmarks/operator_benchmark/c2/add_test.py
imports: import benchmark_caffe2
  from benchmark_caffe2 import Caffe2BenchmarkBase
  from caffe2.python import core
  import operator_benchmark
label_desc: Use image node_id 1 for calling the AddBenchmark obj's underlying member method code with example usage: obj.init(M, N, K, dtype) without return types