Searched refs:expand_tensor_1_shape (Results 1 – 1 of 1) sorted by relevance
1474 …def _helper(self, shape_tensor_1, shape_tensor_2, expand_tensor_1_shape=None, expand_tensor_2_shap… argument1475 if expand_tensor_1_shape:1476 tensor1_mps = torch.randn(shape_tensor_1, device="mps").expand(expand_tensor_1_shape)