code: def _BuildSmallModel(self):
      a = constant_op.constant([[1, 2], [3, 4]])
      b = constant_op.constant([[1, 2], [3, 4]])
      return math_ops.matmul(a, b)
  
file_path: tensorflow/tensorflow/python/profiler/tfprof_logger_test.py
imports: from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the TFProfLoggerTest obj's underlying member method code with example usage: obj._BuildSmallModel() and returns: math_ops