title: Error resulting from Tensorflow prediction fail
body: <p>I have trained a TensorFlow CNN model to perform text classification.  The training completes and evaluates without issue, but when I send predictions to the model when it is hosted on GCP I am getting the following error that I am finding it hard to understand:</p>
  
  <pre><code>RuntimeError: Prediction failed: Error during model execution: AbortionError(code=StatusCode.INVALID_ARGUMENT, details="NodeDef mentions attr 'output_type' not in Op&lt;name=ArgMax; signature=input:T, dimension:Tidx -&gt; output:int64; attr=T:type,allowed=[DT_FLOAT, DT_DOUBLE, DT_INT64, DT_INT32, DT_UINT8, DT_UINT16, DT_INT16, DT_INT8, DT_COMPLEX64, DT_COMPLEX128, DT_QINT8, DT_QUINT8, DT_QINT32, DT_HALF]; attr=Tidx:type,default=DT_INT32,allowed=[DT_INT32, DT_INT64]&gt;; NodeDef: predicted = ArgMax[T=DT_FLOAT, Tidx=DT_INT32, _output_shapes=[[-1]], output_type=DT_INT64, _device="/job:localhost/replica:0/task:0/cpu:0"](fc/logits, predicted/dimension)
       [[Node: predicted = ArgMax[T=DT_FLOAT, Tidx=DT_INT32, _output_shapes=[[-1]], output_type=DT_INT64, _device="/job:localhost/replica:0/task:0/cpu:0"](fc/logits, predicted/dimension)]]")
  </code></pre>
  
  <p>Could anyone tell me what is going wrong here?</p>