def bbox_transform_inv(boxes, deltas, gpu=-1): if gpu >= 0: with cuda.Device(gpu): return _bbox_transform_inv(boxes, deltas) else: return _bbox_transform_inv(boxes, deltas)