| 810 | } |
| 811 | |
| 812 | void SubgraphTask::CreateNodeInputWaitCounter(Node* node) |
| 813 | { |
| 814 | std::atomic<std::uint64_t>* p_counter = new std::atomic<std::uint64_t>(); |
| 815 | node->SetAttr("input_wait_counter", p_counter); |
| 816 | } |
| 817 | |
| 818 | void SubgraphTask::ReleaseNodeInputWaitCounter(Node* node) |
| 819 | { |