Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_mps.py1474 …def _helper(self, shape_tensor_1, shape_tensor_2, expand_tensor_1_shape=None, expand_tensor_2_shap… argument
1475 if expand_tensor_1_shape:
1476 tensor1_mps = torch.randn(shape_tensor_1, device="mps").expand(expand_tensor_1_shape)