| 597 | } |
| 598 | |
| 599 | PDNode *PDNode::assert_is_ops(const std::unordered_set<std::string> &op_types) { |
| 600 | asserts_.emplace_back([op_types](Node *x) { |
| 601 | return x && x->IsOp() && op_types.count(x->Op()->Type()); |
| 602 | }); |
| 603 | return this; |
| 604 | } |
| 605 | |
| 606 | PDNode *PDNode::assert_is_ops_nth_input( |
| 607 | const std::unordered_set<std::string> &op_types, |
no test coverage detected