Lines Matching refs:as_tuple
693 dst1 = torch.nonzero(tensor, as_tuple=False)
694 dst2 = tensor.nonzero(as_tuple=False)
726 tup1 = torch.nonzero(tensor, as_tuple=True)
727 tup2 = tensor.nonzero(as_tuple=True)
738 torch.nonzero(t, as_tuple=True, out=out)
741 torch.nonzero(t, as_tuple=False, out=out), torch.nonzero(t, out=out)
747 tuple_result = torch.nonzero(t, as_tuple=True)
748 nontuple_result = torch.nonzero(t, as_tuple=False)
750 torch.nonzero(t, as_tuple=False, out=out)
759 expected_tuple = torch.nonzero(t, as_tuple=True)
773 dst1 = tensor.nonzero(as_tuple=False)
774 dst2 = tensor_nc.nonzero(as_tuple=False)