message: [TensorRT]: Define a shape function.
  This will help address some of the issues encountered
  when using TensorRT using other language APIs.
  See context in
  #23853#issuecomment-441867084
diff: @@ -16,6 +16,7 @@ limitations under the License.
  #if GOOGLE_CUDA
  #if GOOGLE_TENSORRT
  +#include "tensorflow/core/framework/common_shape_fns.h"
  #include "tensorflow/core/framework/op.h"
  #include "tensorflow/core/framework/op_kernel.h"
  #include "tensorflow/core/framework/shape_inference.h"
  @@ -43,7 +44,7 @@ REGISTER_OP("TRTEngineOp")
  .Attr("calibration_data: string = ''")
  .Attr("use_calibration: bool = true")
  .Input("in_tensor: InT")
  - .Output("out_tensor: OutT");
  + .Output("out_tensor: OutT")
  // TODO(jie): TF requires concrete output shape for concrete input shapes.
  // This is tricky for batch dimension, since we cannot ensure which input
  // would carry the correct batch dimension (for the current stage of the
  @@ -51,7 +52,7 @@ REGISTER_OP("TRTEngineOp")
  // size, but this could change in the future). Hence we disable shape
  // inference function as a workaround.
  // .SetShapeFn(shape_inference::TRTEngineOpShapeInference);
  -
  + .SetShapeFn(shape_inference::UnknownShape);
  } // namespace tensorflow
  #endif // GOOGLE_TENSORRT
  