请问大神，下面的这些操作是什么意思？

[team-704]

因为课题需要，想了解多gpu在使用tensorflow训练，在一篇博客中发现了这个，就是不清楚这个函数定义的意思，
求大神指定一二。也去查了tf.NodeDef的文档，不太懂，求指点

# By default, all variables will be placed on '/gpu:0'
# So we need a custom device function, to assign all variables to '/cpu:0'
# Note: If GPUs are peered, '/gpu:0' can be a faster option
PS_OPS = ['Variable', 'VariableV2', 'AutoReloadVariable']

def assign_to_device(device, ps_device='/cpu:0'):
    def _assign(op):
        node_def = op if isinstance(op, tf.NodeDef) else op.node_def
        if node_def.op in PS_OPS:
            return "/" + ps_device
        else:
            return device

    return _assign