Skip to content

Commit f1dff83

Browse files
committed
support OP cross_entropy_loss, cross_entropy_loss_back
1 parent 4fc4ec5 commit f1dff83

5 files changed

Lines changed: 618 additions & 263 deletions

File tree

docs/ops.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ Legend:
3535
| COS |||||||| 🟡 ||||
3636
| COUNT_EQUAL ||||||||||||
3737
| CPY || 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 |||
38-
| CROSS_ENTROPY_LOSS ||||||| |||||
39-
| CROSS_ENTROPY_LOSS_BACK ||||||| |||||
38+
| CROSS_ENTROPY_LOSS ||||||| |||||
39+
| CROSS_ENTROPY_LOSS_BACK ||||||| |||||
4040
| CUMSUM ||||||||||||
4141
| DIAG ||||||||||||
4242
| DIAG_MASK_INF |||||| 🟡 ||||||
@@ -70,7 +70,7 @@ Legend:
7070
| MUL ||||| 🟡 |||||||
7171
| MUL_MAT | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 | 🟡 |
7272
| MUL_MAT_HADAMARD ||||||||||||
73-
| MUL_MAT_ID || 🟡 ||| 🟡 | 🟡 | 🟡 || 🟡 | 🟡 ||
73+
| MUL_MAT_ID || 🟡 ||| 🟡 | 🟡 | || 🟡 | 🟡 ||
7474
| NEG |||| 🟡 ||||||||
7575
| NORM |||||||| 🟡 ||||
7676
| OPT_STEP_ADAMW ||||||||||||

0 commit comments

Comments
 (0)