| 1027 | } |
| 1028 | |
| 1029 | Status FinishedCallback(int64_t total_num_batches) { |
| 1030 | bool expected = false; |
| 1031 | if (complete_.compare_exchange_strong(expected, true)) { |
| 1032 | return output_->InputFinished(this, static_cast<int>(total_num_batches)); |
| 1033 | } |
| 1034 | return Status::OK(); |
| 1035 | } |
| 1036 | |
| 1037 | private: |
| 1038 | AtomicCounter batch_count_[2]; |
no test coverage detected