Searched refs:shape_and_device_inference (Results 1 – 1 of 1) sorted by relevance
218 … use_factory_function, shape_and_device_inference, input_kind): argument219 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: