| 176 | } |
| 177 | |
| 178 | void SessionMgr::SetLogging(bool active) { |
| 179 | mutex_lock l(mu_); |
| 180 | this->is_logging_active_ = active; |
| 181 | // Legacy Session |
| 182 | if (legacy_session_) { |
| 183 | auto* worker_cache = legacy_session_->worker_cache.get(); |
| 184 | if (worker_cache) { |
| 185 | worker_cache->SetLogging(active); |
| 186 | } |
| 187 | } |
| 188 | |
| 189 | for (const auto& session_kv : sessions_) { |
| 190 | auto session = session_kv.second.get(); |
| 191 | if (session) { |
| 192 | auto* worker_cache = session->worker_cache.get(); |
| 193 | if (worker_cache) { |
| 194 | worker_cache->SetLogging(active); |
| 195 | } |
| 196 | } |
| 197 | } |
| 198 | } |
| 199 | |
| 200 | void SessionMgr::RetrieveLogs(tensorflow::int64 step_id, |
| 201 | LoggingResponse* response) { |
no test coverage detected