Lines Matching refs:TensorPair
1378 class TensorPair: class
1385 return TensorPair(l, r)
1390 return TensorPair(l, r)
1392 def use_tensor_pair(x : TensorPair, y : TensorPair): argument
1396 x = TensorPair(torch.randn(5, 3), torch.randn(5, 3))
1397 y = TensorPair(torch.randn(5, 3), torch.randn(5, 3))
1408 class TensorPair(metaclass=torch.fx.ProxyableClassMeta): class
1415 return TensorPair(l, r)
1420 return TensorPair(l, r)
1422 def use_tensor_pair_literal(x : TensorPair): argument
1423 s = x.add(TensorPair(torch.zeros(5, 3), torch.zeros(5, 3)))
1426 x = TensorPair(torch.randn(5, 3), torch.randn(5, 3))
1437 class TensorPair(metaclass=torch.fx.ProxyableClassMeta): class
1444 return TensorPair(l, r)
1449 return TensorPair(l, r)
1451 def use_tensor_pair_ctor(x : TensorPair, y : torch.Tensor): argument
1452 s = x.add(TensorPair(y, y))
1455 x = TensorPair(torch.randn(5, 3), torch.randn(5, 3))