| 120 | } |
| 121 | |
| 122 | Status GetSize(Tensor* size) { |
| 123 | std::unique_lock<std::mutex> lock(mu_); |
| 124 | size->scalar<int64>().setConstant(static_cast<int64>(queue_.size())); |
| 125 | return Status::OK(); |
| 126 | } |
| 127 | |
| 128 | Status Restore(const Tensor& restorable) { |
| 129 | const int64 num_works = restorable.shape().dim_size(0); |