Home
last modified time | relevance | path

Searched refs:shapes_to_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py152 def shapes_to_tensor(x, device=None): function
224 sizes = shapes_to_tensor([x.__len__() for x in box_lists], device=boxes.device)