f97074b0c0d3a71488f63a01aecf73a703fa571e,official/vision/beta/serving/export_base.py,ExportModule,inference_from_image_tensors,#ExportModule#,64

Before Change

  @tf.function
  def inference_from_image_tensors(self, input_tensor):
    return dict(outputs=self._run_inference_on_image_tensors(input_tensor))

  @tf.function
  def inference_from_image_bytes(self, input_tensor):
    with tf.device("cpu:0"):

After Change

  @tf.function
  def inference_from_image_tensors(self, input_tensor):
    return self._run_inference_on_image_tensors(input_tensor)

  @tf.function
  def inference_from_image_bytes(self, input_tensor):
    with tf.device("cpu:0"):

Instances