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: Check that the expected engine is built.
  
  Args:
    run_params: the run parameters.
  
  Returns:
    the expected engines to build.
  
  The squeeze op is not converted by TensorRT in implicit batch mode.
  Because of this we have two TRTEngineOp in the graphs: one for the
  subgraph before 'squeeze(q,0)', and another one for the rest of the ops
  after the 'squeeze(q,0)'.
  
  In explicit batch mode the whole graph is converted using a single engine.
docstr_tok: ["Check","that","the","expected","engine","is","built",".","Args",":","run_params",":","the","run","parameters",".","Returns",":","the","expected","engines","to","build",".","The","squeeze","op","is","not","converted","by","TensorRT","in","implicit","batch","mode",".","Because","of","this","we","have","two","TRTEngineOp","in","the","graphs",":","one","for","the","subgraph","before","'squeeze","(","q,0",")","'",",","and","another","one","for","the","rest","of","the","ops","after","the","'squeeze","(","q,0",")","'",".","In","explicit","batch","mode","the","whole","graph","is","converted","using","a","single","engine","."]
code: def ExpectedEnginesToBuild(self, run_params):
      """Check that the expected engine is built.
  
      Args:
        run_params: the run parameters.
  
      Returns:
        the expected engines to build.
  
      The squeeze op is not converted by TensorRT in implicit batch mode.
      Because of this we have two TRTEngineOp in the graphs: one for the
      subgraph before 'squeeze(q,0)', and another one for the rest of the ops
      after the 'squeeze(q,0)'.
  
      In explicit batch mode the whole graph is converted using a single engine.
      """
      if run_params.dynamic_shape:
          return ["TRTEngineOp_000"]
      else:
          return ["TRTEngineOp_000", "TRTEngineOp_001"]
  
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 6 for calling the TrtModeTestBase obj's underlying member method code with example usage: obj.ExpectedEnginesToBuild(run_params) and returns: list, list