| 42 | } |
| 43 | |
| 44 | SymbolVar fma3(SymbolVar a, SymbolVar b, SymbolVar c) { |
| 45 | return Elemwise::make({a, b, c}, Mode::FUSE_MUL_ADD3); |
| 46 | } |
| 47 | |
| 48 | SymbolVar fma4(SymbolVar a, SymbolVar b, SymbolVar c, SymbolVar d) { |
| 49 | return Elemwise::make({a, b, c, d}, Mode::FUSE_MUL_ADD4); |