Hello,
the device placement can be specified using tf.device in this way:
with tf.device("/gpu:0"):
    x = tf.Variable(shape, dtpe=tf.int64) # shape is a variable defined somewhere, with the shape

In this case the x variable is placed in the first GPU, and the type is 64 bit integer