class_docstr: Testing the output shape of a TRTEngine.
class_docstr_tok: ["Testing","the","output","shape","of","a","TRTEngine","."]
code: def GraphFn(self, inp):
      b = array_ops.squeeze(inp, axis=[2])
      c = nn.relu(b)
      d1 = c + c
      d2 = math_ops.reduce_sum(d1)
  
      d1 = array_ops.identity(d1, name="output_0")
      d2 = array_ops.identity(d2, name="output_1")
      return d1, d2
  
file_path: tensorflow/tensorflow/python/compiler/tensorrt/test/trt_engine_op_shape_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_shape
  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
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import signature_constants
  from tensorflow.python.saved_model import tag_constants
label_desc: Use image node_id 1 for calling the TRTEngineOpInputOutputShapeTest obj's underlying member method code with example usage: obj.GraphFn(inp) and returns: d1,  d2