model_deploy.py 文件源码

def choose(self, op):
    if op.device:
      return op.device
    node_def = op if isinstance(op, tf.NodeDef) else op.node_def
    if node_def.op in ["Variable", "VariableV2"]:
      return self._var_device
    else:
      return self._clone_device