Function CopyDeviceOption

caffe2/python/core_gradients_test.py:35–38

(op, src_op)

def CopyDeviceOption(op, src_op):
    if src_op.HasField('device_option'):
        op.device_option.CopyFrom(src_op.device_option)
    return op


# First gradient: (in -> out) leading to (out_grad -> in_grad)