class_docstr: Testing BatchMatMulV2: one operand is weight and both have same rank.
class_docstr_tok: ["Testing","BatchMatMulV2",":","one","operand","is","weight","and","both","have","same","rank","."]
code: def GraphFn(self, inp):
      dtype = inp.dtype
      b = constant_op.constant(
          np.random.randn(1, 5, 7), dtype=dtype, name="kernel"
      )
      x1 = math_ops.matmul(inp, b, name="matmul")
      return array_ops.identity(x1, name="output_0")
  
imports: import unittest
  import numpy
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  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.ops import nn
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the BatchMatMulWeightBroadcastTest obj's underlying member method code with example usage: obj.GraphFn(inp) and returns: array_ops