| 798 | } |
| 799 | |
| 800 | std::uint64_t SubgraphTask::GetNodeInputWaitMask(Node* node) |
| 801 | { |
| 802 | if(!node->ExistAttr("input_wait")) |
| 803 | return 0; |
| 804 | return any_cast<std::uint64_t>(node->GetAttr("input_wait")); |
| 805 | } |
| 806 | |
| 807 | std::atomic<std::uint64_t>* SubgraphTask::GetNodeInputWaitCounter(Node* node) |
| 808 | { |