code: def mpi_send_wait_key(a):
      """Wait as long as possible on Waits, Start Send/Recvs early."""
      if isinstance(a.op, (MPIRecvWait, MPISendWait)):
          return 1
      if isinstance(a.op, (MPIRecv, MPISend)):
          return -1
      return 0
  
imports: from __future__ import absolute_import, print_function, division
  import numpy
  from theano import gof
  from theano.gof import Constant, Generic, Op
  from theano.gof.sched import key_to_cmp
  from theano.tensor import tensor
  import theano
label_desc: Use image node_id 6 for calling a global function with example usage: mpi_send_wait_key(a) and returns: int, int, unknown