class_docstr: Base class for BatchMatMult tests.
class_docstr_tok: ["Base","class","for","BatchMatMult","tests","."]
code: def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes):
      return self.BuildParamsWithMask(
          graph_fn=graph_fn,
          dtype=dtype,
          input_shapes=input_shapes,
          output_shapes=output_shapes,
          input_mask=[[True] * len(s) for s in input_shapes],
          output_mask=[[True] * len(s) for s in output_shapes],
          extra_inputs=[],
          extra_outputs=[],
      )
  
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 1 for calling the BatchMatMultTestBase obj's underlying member method code with example usage: obj.BuildParams(graph_fn, dtype, input_shapes, output_shapes) and returns: self