Home
last modified time | relevance | path

Searched defs:BoxedDeviceIndex (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcompile_fx.py447 boxed_forward_device_index: Optional[BoxedDeviceIndex] = None,
1186 forward_device: BoxedDeviceIndex,
H A Dcudagraph_utils.py193 class BoxedDeviceIndex: class
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py1786 boxed_forward_device_index: Optional[BoxedDeviceIndex] = None,