|
20 | 20 | 融合算子 |
21 | 21 | 神经网络层:Conv, ConvInteger, ConvTranspose, DeformConv, GRU, LSTM, RNN, QLinearConv, QLinearMatMul |
22 | 22 | 损失函数:NegativeLogLikelihoodLoss, SoftmaxCrossEntropyLoss |
23 | | -其他融合操作:AffineGrid, CenterCropPad, Col2Im, Compress, DFT, ImageDecoder, Loop, NonMaxSuppression, Optional, OptionalGetElement, OptionalHasElement, RegexFullMatch, Scan, TfIdfVectorizer, Upsample |
| 23 | +其他融合操作:AffineGrid, CenterCropPad, Col2Im, Compress, DFT, ImageDecoder, Loop, NonMaxSuppression, Optional, OptionalGetElement, OptionalHasElement, RegexFullMatch, Scan, TfIdfVectorizer, Upsample |
| 24 | + |
| 25 | +## op通讯指令格式 |
| 26 | + |
| 27 | +### 设置常量 |
| 28 | ++ int数组 |
| 29 | +argset@int 1 2 3->v_1 |
| 30 | ++ int |
| 31 | +argset@int 10->int_10 |
| 32 | ++ float数组 |
| 33 | +argset@float 1.0 2.0 3.0->v_3 |
| 34 | ++ float |
| 35 | +argset@float 10.0->float_10 |
| 36 | + |
| 37 | + |
| 38 | + |
| 39 | +### 创建张量 |
| 40 | +newtensor@int32 1 2 3->a |
| 41 | + |
| 42 | +### 填充张量 |
| 43 | + |
| 44 | +constant@int32 int_10->a |
| 45 | + |
| 46 | +### 打印张量 |
| 47 | +print a |
| 48 | + |
| 49 | +### 逐元素操作 |
| 50 | +add@float32 a b->a requires_grad=true |
| 51 | +add_scalar@float32 a 10->a requires_grad=true |
| 52 | +mul@float32 a(a_grad) b(b_grad)->a(a_grad) requires_grad=true |
| 53 | +mul_scalar@float32 a 10->a requires_grad=true |
| 54 | +div@float32 a b->a requires_grad=true |
| 55 | +div_scalar@float32 a 10->a requires_grad=true |
| 56 | +sqrt@float32 a->a requires_grad=true |
| 57 | +exp@float32 a->a requires_grad=true |
| 58 | + |
| 59 | + |
| 60 | +### 广播操作 |
| 61 | +broadcast@float32 a b->a requires_grad=true |
| 62 | + |
| 63 | + |
| 64 | +### 矩阵操作 |
| 65 | +matmul@float32 a b->a requires_grad=true |
| 66 | + |
| 67 | + |
| 68 | +### 归约操作 |
| 69 | +sum@float32 a->a requires_grad=true |
| 70 | +max@float32 a->a requires_grad=true |
| 71 | +min@float32 a->a requires_grad=true |
| 72 | + |
| 73 | + |
| 74 | +### 形状操作 |
| 75 | +reshape@float32 a 1 2 3->a requires_grad=true |
| 76 | +flatten@float32 a->a requires_grad=true |
| 77 | +transpose@float32 a 1 2->a requires_grad=true |
| 78 | +unsqueeze@float32 a 1->a requires_grad=true |
| 79 | +squeeze@float32 a 1->a requires_grad=true |
| 80 | + |
| 81 | +### 类型操作 |
| 82 | +cast@float32 a->a requires_grad=true |
| 83 | +cast_like@float32 a b->a requires_grad=true |
| 84 | + |
| 85 | + |
| 86 | + |
| 87 | + |
| 88 | + |
| 89 | + |
| 90 | + |
| 91 | + |
| 92 | + |
0 commit comments