Searched refs:shapes_to_tensor (Results 1 – 1 of 1) sorted by relevance
152 def shapes_to_tensor(x, device=None): function224 sizes = shapes_to_tensor([x.__len__() for x in box_lists], device=boxes.device)