Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py218 … use_factory_function, shape_and_device_inference, input_kind): argument
219 if input_kind == 'list' and shape_and_device_inference:
229 …evice(device).type == 'cuda' and torch.cuda.device_count() >= 2 and not shape_and_device_inference:
255 if shape_and_device_inference and 0 in size:
274 if shape_and_device_inference:
282 if shape_and_device_inference: