from tensorflow.contrib.learn.python.learn.graph_actions import infer

        predictions = {'predictions': predictions}
      # TODO(ipolosukhin): Support batching
      return infer(checkpoint_path, predictions, feed_dict=feed_dict)

