def get_device(self, unused_tree_num):
    if (not self.cached):
        dummy = constant_op.constant(0)
        self.cached = dummy.device
    return self.cached
def get_device(self, unused_tree_num):
    if (not self.cached):
        dummy = constant_op.constant(0)
        self.cached = dummy.device
    return self.cached