public static final int
OP_ADAMW_UPDATE
20
public static final int
OP_ADD
1
public static final int
OP_ADD_BROADCAST_COLS
25
public static final int
OP_ADD_ROW_VECTOR
24
public static final int
OP_ADD_SCALAR
21
public static final int
OP_BATCHED_MATMUL
44
public static final int
OP_CAUSAL_MASK
47
public static final int
OP_CAUSAL_SOFTMAX
50
public static final int
OP_CLAMP
35
public static final int
OP_CROSS_ENTROPY_FUSED
51
public static final int
OP_CROSS_ENTROPY_GRADIENT
40
public static final int
OP_CROSS_ENTROPY_LOSS
39
public static final int
OP_DIVIDE
4
public static final int
OP_DIVIDE_BROADCAST_COLS
27
public static final int
OP_DIVIDE_SCALAR
22
public static final int
OP_EXP
9
public static final int
OP_GATHER_ROWS
49
public static final int
OP_GELU
15
public static final int
OP_GELU_BACKWARD
16
public static final int
OP_HEAD_CACHE_WRITE
60
public static final int
OP_HEAD_PREFIX_MATMUL
61
public static final int
OP_LAYERNORM_BACKWARD
19
public static final int
OP_LAYOUT_ADAMW
76
public static final int
OP_LAYOUT_ASSIGN
66
public static final int
OP_LAYOUT_BINARY
64
public static final int
OP_LAYOUT_CAUSAL
78
public static final int
OP_LAYOUT_CROSS_ENTROPY
77
public static final int
OP_LAYOUT_INDEX_ADD
72
public static final int
OP_LAYOUT_INDEX_SELECT
71
public static final int
OP_LAYOUT_MATMUL
70
public static final int
OP_LAYOUT_MAXIMUM_BACKWARD
69
public static final int
OP_LAYOUT_NORMALIZATION_BACKWARD
74
public static final int
OP_LAYOUT_NORMALIZE
73
public static final int
OP_LAYOUT_REDUCE
67
public static final int
OP_LAYOUT_ROTARY
79
public static final int
OP_LAYOUT_SOFTMAX
68
public static final int
OP_LAYOUT_SWIGLU_BACKWARD
75
public static final int
OP_LAYOUT_TRANSFER
63
public static final int
OP_LAYOUT_UNARY
65
public static final int
OP_LOG
10
public static final int
OP_MATMUL
5
public static final int
OP_MAX_ALONG_ROWS
34
public static final int
OP_MEAN_ALONG_ROWS
30
public static final int
OP_MERGE_HEADS
46
public static final int
OP_MULTIPLY
3
public static final int
OP_MULTIPLY_BROADCAST_COLS
32
public static final int
OP_MULTIPLY_BROADCAST_ROWS
28
public static final int
OP_MULTIPLY_SCALAR
6
public static final int
OP_NEG
8
public static final int
OP_POW
36
public static final int
OP_RELU
13
public static final int
OP_RELU_BACKWARD
14
public static final int
OP_RMS_NORM
53
public static final int
OP_RMS_NORM_BACKWARD
54
public static final int
OP_ROTARY
48
public static final int
OP_SCALED_SOFTMAX_BACKWARD
57
public static final int
OP_SCATTER_ADD_ROWS
37
public static final int
OP_SCATTER_ADD_ROWS_ATOMIC
42
public static final int
OP_SIGMOID
12
public static final int
OP_SOFTMAX_BACKWARD
18
public static final int
OP_SOFTMAX_ROWS
17
public static final int
OP_SPLIT_HEADS
45
public static final int
OP_SQRT
7
public static final int
OP_STORAGE_COPY
62
public static final int
OP_SUBTRACT
2
public static final int
OP_SUBTRACT_BROADCAST_COLS
26
public static final int
OP_SUM_ABS
38
public static final int
OP_SUM_ALONG_ROWS
33
public static final int
OP_SUM_ROWS
29
public static final int
OP_SUM_SCALAR
41
public static final int
OP_SUM_SQUARES
43
public static final int
OP_SUM_SQUARES_SCALAR
52
public static final int
OP_SWIGLU
55
public static final int
OP_SWIGLU_BACKWARD
56
public static final int
OP_TANH
11
public static final int
OP_TILED_CAUSAL_ATTENTION
58
public static final int
OP_TILED_CAUSAL_ATTENTION_BACKWARD
59
public static final int
OP_TRANSPOSE
23
public static final int
OP_VARIANCE_ALONG_ROWS
31