1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
from typing import Optional, Tuple
import torch
def cuda_version_placeholder() -> int:
return -1
def scatter_placeholder(src: torch.Tensor, index: torch.Tensor, dim: int,
out: Optional[torch.Tensor],
dim_size: Optional[int]) -> torch.Tensor:
raise ImportError
return src
def scatter_arg_placeholder(src: torch.Tensor, index: torch.Tensor, dim: int,
out: Optional[torch.Tensor],
dim_size: Optional[int]
) -> Tuple[torch.Tensor, torch.Tensor]:
raise ImportError
return src, index
def segment_csr_placeholder(src: torch.Tensor, indptr: torch.Tensor,
out: Optional[torch.Tensor]) -> torch.Tensor:
raise ImportError
return src
def segment_csr_arg_placeholder(src: torch.Tensor, indptr: torch.Tensor,
out: Optional[torch.Tensor]
) -> Tuple[torch.Tensor, torch.Tensor]:
raise ImportError
return src, indptr
def gather_csr_placeholder(src: torch.Tensor, indptr: torch.Tensor,
out: Optional[torch.Tensor]) -> torch.Tensor:
raise ImportError
return src
def segment_coo_placeholder(src: torch.Tensor, index: torch.Tensor,
out: Optional[torch.Tensor],
dim_size: Optional[int]) -> torch.Tensor:
raise ImportError
return src
def segment_coo_arg_placeholder(src: torch.Tensor, index: torch.Tensor,
out: Optional[torch.Tensor],
dim_size: Optional[int]
) -> Tuple[torch.Tensor, torch.Tensor]:
raise ImportError
return src, index
def gather_coo_placeholder(src: torch.Tensor, index: torch.Tensor,
out: Optional[torch.Tensor]) -> torch.Tensor:
raise ImportError
return src
|
