class_docstr: Testing that output type of engine using Top-K is set correctly.
code: def GraphFn(self, x):
      k = 5
      k_tensor = constant_op.constant(
          k, dtype=dtypes.int32, name="Const"
      )
      values, indices = nn_ops.top_k(x, k_tensor, name="TopK")
      # Reshape will act as a layer between the TopK output and the engine
      # output, requiring the output tensor of reshape to be set explicitly to
      # int32.
      indices = array_ops.reshape(indices, [100, 1, 5], name="Reshape")
      values = array_ops.identity(values, name="output_0")
      indices = array_ops.identity(indices, name="output_1")
      return values, indices
  
file_path: tensorflow/tensorflow/python/compiler/tensorrt/test/topk_test.py
imports: from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import constant_op
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the TopKOutputTypeTest obj's underlying member method code with example usage: obj.GraphFn(x) and returns: values,  indices