| 805 | } |
| 806 | |
| 807 | std::atomic<std::uint64_t>* SubgraphTask::GetNodeInputWaitCounter(Node* node) |
| 808 | { |
| 809 | return any_cast<std::atomic<std::uint64_t>*>(node->GetAttr("input_wait_counter")); |
| 810 | } |
| 811 | |
| 812 | void SubgraphTask::CreateNodeInputWaitCounter(Node* node) |
| 813 | { |