Home
last modified time | relevance | path

Searched refs:TraceMeConsumer (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/lib/
Dconnected_traceme.h105 class TraceMeConsumer {
108 TraceMeConsumer(NameT name, ContextType context_type, uint64 context_id,
117 TraceMeConsumer(NameT name, uint64 context_id, int level = 2)
118 : TraceMeConsumer(name, ContextType::kGeneric, context_id, level) {} in TraceMeConsumer() function
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc62 profiler::TraceMeConsumer compute_activity( in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc666 profiler::TraceMeConsumer activity( in Process()
1287 profiler::TraceMeConsumer activity( in Finish()
1310 profiler::TraceMeConsumer activity( in Finish()
1325 profiler::TraceMeConsumer activity( in Finish()
Dbase_collective_executor.cc330 profiler::TraceMeConsumer consumer( in ExecuteAsync()
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dadaptive_shared_batch_scheduler.h535 profiler::TraceMeConsumer trace_me( in CallbackWrapper()
Dshared_batch_scheduler.h815 profiler::TraceMeConsumer trace_me( in ProcessBatch()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc494 profiler::TraceMeConsumer activity( in ExecuteAsync()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc727 tensorflow::profiler::TraceMeConsumer traceme("Run Collective", in LoopKernelLaunches()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1691 tensorflow::profiler::TraceMeConsumer activity( in EnqueueExecution()