| 99 | } |
| 100 | |
| 101 | Status XlaTensor::RefreshStatusOfStreams() { |
| 102 | mutex_lock lock(mu_); |
| 103 | Status status; |
| 104 | for (se::Stream* stream : streams_defined_on_) { |
| 105 | status.Update(stream->RefreshStatus()); |
| 106 | } |
| 107 | return status; |
| 108 | } |
| 109 | |
| 110 | // The pointer tag, OR-ed into the XlaTensor's address to distinguish it from |
| 111 | // device-side tensors, which are either CPU or GPU memory pointers. This works |
no test coverage detected