class_docstr: Test squeeze on batch dim and some unary operations in TF-TRT.
class_docstr_tok: ["Test","squeeze","on","batch","dim","and","some","unary","operations","in","TF-TRT","."]
docstr: The input has 1 as a first dimension, which is removed by the squeeze.
  
  op in the graph.
  
  In explicit batch mode, TensorRT can convert the whole graph. In this mode
  it is possible to manipulate the batch dimension using the squeeze op.
  
  In implicit batch mode TensorRT cannot convert the whole graph. We are not
  allowed to manipulate (squeeze) the first dimension in implicit batch mode.
  Therefore the graph will be converted using multiple segments.
docstr_tok: ["The","input","has","1","as","a","first","dimension",",","which","is","removed","by","the","squeeze",".","op","in","the","graph",".","In","explicit","batch","mode",",","TensorRT","can","convert","the","whole","graph",".","In","this","mode","it","is","possible","to","manipulate","the","batch","dimension","using","the","squeeze","op",".","In","implicit","batch","mode","TensorRT","can","not","convert","the","whole","graph",".","We","are","not","allowed","to","manipulate","(","squeeze",")","the","first","dimension","in","implicit","batch","mode",".","Therefore","the","graph","will","be","converted","using","multiple","segments","."]
code: def GetParams(self):
      """The input has 1 as a first dimension, which is removed by the squeeze.
  
      op in the graph.
  
      In explicit batch mode, TensorRT can convert the whole graph. In this mode
      it is possible to manipulate the batch dimension using the squeeze op.
  
      In implicit batch mode TensorRT cannot convert the whole graph. We are not
      allowed to manipulate (squeeze) the first dimension in implicit batch mode.
      Therefore the graph will be converted using multiple segments.
      """
      return self.BuildParams(
          self.GraphFn, dtypes.float32, [[1, 12, 5]], [[12, 5]]
      )
  
file_path: tensorflow/tensorflow/python/compiler/tensorrt/test/trt_mode_test.py
imports: from unittest import SkipTest
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  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 3 for calling the TrtModeTestBase obj's underlying member method code with example usage: obj.GetParams() and returns: self