| 6 | |
| 7 | namespace { |
| 8 | bool is_opr_memforward_var(mgb::VarNode* var) { |
| 9 | if (var) { |
| 10 | auto opr = var->owner_opr(); |
| 11 | if (opr->try_cast_final<mgb::opr::Reshape>() || |
| 12 | opr->try_cast_final<mgb::opr::Broadcast>() || |
| 13 | opr->try_cast_final<mgb::opr::Subtensor>() || |
| 14 | opr->try_cast_final<mgb::opr::AxisAddRemove>() || |
| 15 | opr->try_cast_final<mgb::opr::Dimshuffle>()) { |
| 16 | return true; |
| 17 | } |
| 18 | }; |
| 19 | return false; |
| 20 | } |
| 21 | } // namespace |
| 22 | |
| 23 | namespace mgb { |
no test coverage detected