| 1550 | } |
| 1551 | |
| 1552 | Maybe<void> CheckOpInputSignature(const Operator& op, const OpNodeSignature& upstream_signature) { |
| 1553 | for (const auto& ibn : op.input_bns()) { |
| 1554 | { |
| 1555 | CHECK_OR_RETURN(upstream_signature.has_logical_blob_desc_signature()); |
| 1556 | const auto& map = upstream_signature.logical_blob_desc_signature().bn_in_op2blob_desc(); |
| 1557 | CHECK_OR_RETURN(map.find(ibn) != map.end()); |
| 1558 | } |
| 1559 | { |
| 1560 | CHECK_OR_RETURN(upstream_signature.has_sbp_signature()); |
| 1561 | const auto& map = upstream_signature.sbp_signature().bn_in_op2sbp_parallel(); |
| 1562 | CHECK_OR_RETURN(map.find(ibn) != map.end()); // NOLINT |
| 1563 | } |
| 1564 | { |
| 1565 | CHECK_OR_RETURN(upstream_signature.has_local_signature()); // NOLINT |
| 1566 | const auto& map = upstream_signature.local_signature().bn_in_op2opt_local_parallel(); |
| 1567 | CHECK_OR_RETURN(map.find(ibn) != map.end()); |
| 1568 | } |
| 1569 | } |
| 1570 | return Maybe<void>::Ok(); |
| 1571 | } |
| 1572 | |
| 1573 | } // namespace |
| 1574 |
no test coverage detected