Commit fbc7f1e
authored
[Relax] Replace topi.take with relax.op.take (#18665)
## Why
TODO comment indicated that topi.take should be replaced with
relax.op.take once it has better support. The relax take operator is now
mature enough to replace the TOPI external call.
## How
- Replace bb.emit_te(topi.take, ...) with relax.op.take(..., axis=0) in
EmbedLayerNormalization converter1 parent c866abc commit fbc7f1e
1 file changed
Lines changed: 3 additions & 4 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
3195 | 3195 | | |
3196 | 3196 | | |
3197 | 3197 | | |
3198 | | - | |
3199 | | - | |
| 3198 | + | |
3200 | 3199 | | |
3201 | | - | |
3202 | | - | |
| 3200 | + | |
| 3201 | + | |
3203 | 3202 | | |
3204 | 3203 | | |
3205 | 3204 | | |
| |||
0 commit comments