diff options
Diffstat (limited to 'tensorflow/core/common_runtime/direct_session.cc')
| -rw-r--r-- | tensorflow/core/common_runtime/direct_session.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/direct_session.cc b/tensorflow/core/common_runtime/direct_session.cc index 458e133b68..52c1cd2691 100644 --- a/tensorflow/core/common_runtime/direct_session.cc +++ b/tensorflow/core/common_runtime/direct_session.cc @@ -64,6 +64,7 @@ limitations under the License. #include "tensorflow/core/platform/device_tracer.h" #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/mutex.h" +#include "tensorflow/core/platform/tracing.h" #include "tensorflow/core/platform/types.h" #include "tensorflow/core/util/device_name_utils.h" #include "tensorflow/core/util/env_var.h" @@ -453,6 +454,9 @@ Status DirectSession::RunInternal(int64 step_id, const RunOptions& run_options, CallFrameInterface* call_frame, ExecutorsAndKeys* executors_and_keys, RunMetadata* run_metadata) { + string session_id_meta = strings::StrCat("SessionRun #id=", step_id, "#"); + tracing::ScopedActivity activity(session_id_meta); + const int64 executor_step_count = executors_and_keys->step_count.fetch_add(1); std::unique_ptr<DebuggerStateInterface> debugger_state; |